【Pytorch】如何正确安装pytorch-gpu

最新推荐文章于 2024-01-05 20:10:10 发布

weixin_46111444

最新推荐文章于 2024-01-05 20:10:10 发布

阅读量1.7k

点赞数

文章标签： pytorch 深度学习机器学习

本文链接：https://blog.csdn.net/weixin_46111444/article/details/122665704

版权

一开始是想要在windows上训练yolov5模型，由于本人第一次接触cuda以及pytorch，在安装过程中有各种各样的问题，重装了无数次nvcuda还是无济于事，最后竟然发现一直下载错了版本，故发文记录并且把成功经验分享给大家。

首先本人使用的版本是Anaconda+Pytorch1.8.1+torchvision0.9.0，cuda10.2，显卡为GTX1060，驱动版本441.22
一开始在网上查找各种安装教程，无一例外全部失败了，最后输入torch.cuda.is_available()都是显示False，如图：
在这里插入图片描述
—————————————————————————————————————————————

一开始安装pytorch并没有显示后面有cpu三个字母，直到最后通过pytorch清华源和官网安装教程，突然就在后面加了cpu，由此可见是下载错了版本。
在这里插入图片描述
然后我就在清华源下载了pytorch1.6.0+torchvision0.7.0，使用的是pip，最后发现居然可以打印出True了

可以看到显示了GTX1060了

但是如果使用同一个pip指令从清华源下载pytorch1.8.0（1.6.0版本进行yolov5训练会报错，虽然有解决办法但是强迫症），下载下来的却又不是gpu版本

所以我前往官网查看详情如下：
在这里插入图片描述
上面的指令显示 torch==1.8.1+cu102，所以我知道是源的问题了，可能在2022年pytorch下载方式又变了，而我看的大多数帖子都是旧版，默认是下载gpu版本，才导致一直打印出False。
用官网的指令下载速度挺快的，在这里贴出来https://pytorch.org/get-started/previous-versions/，进入后查看适合自己的cuda版本，记得加上“torch==XXX+cuXXX”即可，至此问题解决，也成功用上1.8.1版本的torch-gpu 在这里插入图片描述
——————————————————————————————————
遇到的坑：
1、如果你也是GTX1060，那么你下载CUDA10.0大概率会出现显卡驱动使用不了的情况，只需在下面的链接下载CUDA10.2+441.22版本即可解决

https://developer.download.nvidia.cn/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_441.22_win10.exe

2、如果你安装成功了，使用tensorflow会出现找不到cudart64_101.dll的报错，只需要进入C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin，将cudart64_102.dll复制多一份，改名为cudart64_101.dll即可，还不行的话就试试像我这样把文件复制改名移动
在这里插入图片描述

cudnn64_8.dll是cuda10.1的dll，把cuda64_7复制改名成8即可。