CUDA+CUDNN+PYTORCH-GPU环境安装

CUDA+CUDNN+PYTORCH-GPU

一部血泪史 安了一天 踩了好多坑 。。。 清华源是什么垃圾 最后还是pip靠谱

step1

首先安装 CUDA

安了我一天 先看一下版本 一定要下符合自己的版本的!!!

然后要先去 GeForce Experience 更新驱动 。。。

step2

就按网上说的 下来后 复制到cuda的文件夹里

step3

最复杂的

由于我是用了conda

网上配置清华源在下最新版本的pytorch时是不中用的

我们还需要加几个源

1
2
3
4
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2

下面这个一定要加

1
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

然后可以参考一下关于anaconda安装torch-gpu的最新方法(超详细,2020最新版) - 知乎 (zhihu.com)

1
conda install pytorch torchvision torchaudio cudatoolkit=11.6

然后还是不行 的我一脸懵逼

conda list 一下

image-20220518113627873

发现还是GPU

fuck

1
conda install https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.11.0-py3.7_cuda11.5_cudnn8_0.tar.bz2

直接去源网站里找 fuck

bug

(62条消息) conda安装GPU版pytorch,却是cpu版本的原因_廿十七的博客-CSDN博客_conda安装pytorch是cpu版本

1
conda uninstall pytorch-mutex
1
conda install https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.9.1-py3.9_cuda11.1_cudnn8_0.tar.bz2
1
conda install https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/torchvision-0.12.0-py39_cu115.tar.bz2
1
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html --target E:\conda\envs\torch1.9\Lib\site-packages

离线下载

先下好whl

然后进入当前目录

1
pip install xxx.whl

利用pkgs里的缓存文件

进入pkgs然后

1
conda install ...

step4

先直接把whl pytorch下载下来

然后 利用离线下载搞他

最终成功