当前位置: 代码迷 >> 综合 >> 老电脑安装pytorch一直显示Your installed CUDA driver is:10.0
  详细解决方案

老电脑安装pytorch一直显示Your installed CUDA driver is:10.0

热度:96   发布时间:2023-10-19 19:05:38.0

Your installed CUDA driver is:10.0

出现这个情况,就是电脑配置最高位10.0

而pytorch-gpu版本,支持10.0的只有pytorch1.2版本

高版本是不支持较低cuda的,所以只能自己在重新找个合适的版本下载了,

个人试过1.6、1.7版本的pytorch,cuda9.2都是没用的,pip install下载后还是无法调用gpu

开始下载pytorch1.2版本cuda10.0

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

老电脑安装pytorch一直显示Your installed CUDA driver is:10.0

安装完成后测试一下

python
import torch
print(torch.__version__)
print(torch.cuda.is_available())

老电脑安装pytorch一直显示Your installed CUDA driver is:10.0

好像是可以正常使用了,但是我们生活中大多时候需要的torchvision都需要大于0.4版本的,不然会报错,这时候下周pytorch1.5.1版本的cuda9.2完美解决这个问题

conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2 -c pytorch

剩下操作和上面一样

  相关解决方案