深度学习cuda和pytorch安装

前言

        最近在学习深度学习，在安装cuda和pytorch遇到许多坑，以下是安装过程。

---

提示：以下是本篇文章正文内容，下面案例可供参考

一、安装cuda10.2

下载链接：CUDA Toolkit Archive | NVIDIA Developer

 不要下太新的版本，pytorch对应的版本还没有这么高

如果点进去可能会出现一下界面

 直接去百度网盘下链接：https://pan.baidu.com/s/1-ELPAlRJTd2CtSTamBjuXw 
提取码：3ybx

默认安装在c盘，建议自定义安装到其他盘。

然后在cmd中输入nvcc  -V

说明安装成功。 

然后去官网下载对应的pytorch，链接：PyTorch

 我这里选择用pip安装，在win＋R输入cmd，在输入命令

pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio===0.10.1+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

 可能下载有点慢，建议下载whl文件，在网盘里面都有，

 链接：https://pan.baidu.com/s/1-ELPAlRJTd2CtSTamBjuXw 
提取码：3ybx

分别在cmd输入pip3 install torch-1.10.1+cu102-cp38-cp38-win_amd64.whl，

                        pip3 install torchvision-0.11.2+cu102-cp38-cp38-win_amd64.whl

                        pip3 install torchaudio-0.10.1+cu102-cp38-cp38-win_amd64.whl

 如果出现报错，把这几个文件放在c盘用户目录下面就行了。

验证pytorch能否调用GPU

import torch
print(torch.cuda.is_available())

返回True说明可以调用

---

总结

如果安装调用不了，建议将cuda和pytorch删掉，找到相对应的版本才能成功！