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
安装完成后测试一下
python
import torch
print(torch.__version__)
print(torch.cuda.is_available())
好像是可以正常使用了,但是我们生活中大多时候需要的torchvision都需要大于0.4版本的,不然会报错,这时候下周pytorch1.5.1版本的cuda9.2完美解决这个问题
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2 -c pytorch
剩下操作和上面一样