如何解决在cuda上安装torch后torch.cuda.is_available()返回False

conda install pytorch torchvision cudatoolkit=10.1
安装完验证是否安装成功:

如何解决在cuda上安装torch后torch.cuda.is_available()返回False

发现import是没有问题的,但是在运行torch.cuda.is_available()之后返回值却是False!!!
解决办法:
先在这里下载自己服务器对应cuda版本的torch和torchvision版本文件
我下载的是这两个文件:

如何解决在cuda上安装torch后torch.cuda.is_available()返回False

把这两个文件传输到服务器上anaconda3/pkgs文件夹下:

如何解决在cuda上安装torch后torch.cuda.is_available()返回False

打开终端,进入该文件夹下,运行以下代码;
pip install 文件1

pip install 文件2
运行之后,可以看到两个命令分别成功地安装上了torch 和torchvision:

如何解决在cuda上安装torch后torch.cuda.is_available()返回False

如何解决在cuda上安装torch后torch.cuda.is_available()返回False

重装torch 和torchvision之后,验证一下torch.cuda.is_available()的返回值。

如何解决在cuda上安装torch后torch.cuda.is_available()返回False

Perfect!

Original: https://blog.csdn.net/AI_BING_/article/details/124969188
Author: 我是好人BING
Title: 如何解决在cuda上安装torch后torch.cuda.is_available()返回False

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/711120/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球