先测试:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果输出的结果是False,那么说明当前的Pytorch版本无法使用显卡。
解决办法:
(1)如果有显卡资源并需要使用显卡资源,先检查一下当前的CUDA版本,在终端中输入
cat /usr/local/cuda/version.txt
(2)然后根据自己的CUDA版本在Pytorch官网查看并安装可用的版本,参考链接如下:
CUDA 10.2
https://download.pytorch.org/whl/cu102/torch_stable.html
CUDA 10.1
https://download.pytorch.org/whl/cu101/torch_stable.html
CUDA 10.0
https://download.pytorch.org/whl/cu100/torch_stable.html
CUDA 9.0
https://download.pytorch.org/whl/cu90/torch_stable.html
CPU及所有CUDA版本
网友评论