使用官方命令直接安装pytorch,并且该命令自动下载了CUDA10.2,在nvidia-smi命令下,也可以看到cuda版本是10.2,但为什么import torch后测试cuda,返回的是false呢?
你的计算机安装的是nvidia受兼容的显示卡么?GTX650或者以上?安装显示卡的驱动了么?
建议通过anaconda安装。首先按照以下链接安装anaconda:
https://blog.csdn.net/qq_15192373/article/details/81091098
然后创建python虚拟环境:
conda create -n pytorch
该命令可自动配置好pytorch所需的所有环境,之后只需激活:
source activate pytorch
然后正常使用python即可。