安装pytorch-gpu版

pytorch GPU版本安装_尘世猫的博客-优快云博客_pytorchgpu版本

目录

首先，安装cuda

然后安装pytorch

---

之前在清华源下载的pytorch是cpu版的 在python下测试torch.cuda.is_available()返回的是false

故在万能的Google下，找到了相关文章，进行整理

首先，安装cuda

没有英伟达控制面板 建议下载一个

然后到官网去下载，我的cuda版本是11.4 ，目测可以下载11版本的，目前暂不知道version那一栏后面的server的意思（但因为看不懂这上面version的意思，我按照我看的pytorch视频来安装了version为10的版本，目前暂未发现问题）

 也可以去CUDA Toolkit 11.4 Downloads | NVIDIA Developer这里下载对应版本

在网上看到还要安装cuDNN，但是我没安装，后续出现问题，再补回来（也可能安装了我不知道，萌新脸）

参考博客：Win10 下安装 CUDA Toolkit ( for PyTorch )

参考博客：win10安装CUDA和cuDNN的正确姿势 - 知乎

然后安装pytorch

注意，这里不要用清华源的下载连接

如果清华源可以的话，还是建议通过清华源下载，毕竟速度快

附清华源下载教程

否则你看conda-list或者pip list时，会发现cpu only或带有cpu字样的数据

建议本地安装，在线安装可能会有网络错误等等

去官网本地安装包进行下载，如果进不去，那就是被那啥了，需要那啥可以私信我（可能不及时回）

然后我听网上说下载速度慢，但是我下载速度挺快的，估计是代理的缘故，所以慢慢等待下载好吧。

一开始进去那个本地安装包网址，可能发现里面的版本之，不知道下载什么，我是通过先

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

然后他给我报错我发现我要的安装包是这个

 （如果他没报错，你就慢慢等他下载完就行，可能是我网络波动，或者你多试试几次）

然后就利用ctrl+F寻找

 把名字对上就行，不知道的话就去官网看版本，然后下载对应版本的软件，目前不知道后缀cp36/cp37/cp38有什么区别，所以只能一个个试了（所以能用pip在线安装就在线安装吧）

 等他下完后，在你的虚拟环境中输入 pip install 你的下载文件的地址

报错的话，貌似没影响，或者你可以去你的虚拟环境的文件envs\里把安全权限改一下

 

 然后弄完三个后pytorch、torchvision、torchaudio即可。

 

torch.cuda.is_available() 

大功告成！

参考博客：torch.cuda.is_available()返回false——解决办法_Nefu_lyh的博客-程序员宅基地 - 程序员宅基地

参考博客：解决torch.cuda.is_available()一直返回False的玄学方法之一_dysljxjsq的博客-优快云博客

参考博客：

Anaconda下pip离线安装Pytorch(亲测可用）_caracol的博客-优快云博客_离线安装pytorch