安装Pytorch后torch.cuda.is_available()返回False问题解决

首先说几点安装Pytorch时需要注意的点:
1.如果有能力,最好在科学上网后,利用类似如下命令直接安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.3
这里注意:上述命令行最后没有-c pytorch,这点与官网的不一样,官网给出的是这样的:

 加上-c pytorch的意思好像是从清华源下载~~
2.如果能力不足,无法科学上网,则需要手动切换镜像源了,比如切换清华源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn
之后仍然利用下述命令安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.3
注意:这里还是没有 -c pytorch。因为我加上 -c pytorch后发现安装会有问题,所以还是去掉 -c pytorch。
3.如果要安装GPU版本的Pytorch,则需要你的电脑上有NVI

安装Pytorch后torch.cuda.is_available()返回False问题解决最先出现在Python成神之路

版权声明:
作者:感冒的梵高
链接:https://www.techfm.club/p/4963.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>