CUDA+CUDNN+PYTORCH-GPU环境安装
CUDA+CUDNN+PYTORCH-GPU
一部血泪史 安了一天 踩了好多坑 。。。 清华源是什么垃圾 最后还是pip靠谱
step1
首先安装 CUDA
安了我一天 先看一下版本 一定要下符合自己的版本的!!!
然后要先去 GeForce Experience 更新驱动 。。。
step2
就按网上说的 下来后 复制到cuda的文件夹里
step3
最复杂的
由于我是用了conda
网上配置清华源在下最新版本的pytorch时是不中用的
我们还需要加几个源
1 | conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free |
下面这个一定要加
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
一下
发现还是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下载下来
然后 利用离线下载搞他
最终成功