前言:前前后后安了两个周,所以暂时还没安装成功的朋友不要着急,再慢也慢不过我了。
大概流程(以QA形式):
1.该安装什么版本的CUDA
https://blog.csdn.net/qq_38295511/article/details/89223169
2. CUDA下载
https://developer.nvidia.com/cuda-toolkit-archive
3. CUDA安装
正常安装(这里应该没什么大坑)
4. Cudnn下载(计算加速用,容易下载,容易配置,没道理不装)
https://developer.nvidia.com/rdp/cudnn-archive
5. Cudnn配置
https://blog.csdn.net/weixin_38673554/article/details/90513732
6. Anaconda下载
官网下载最新版:
https://www.anaconda.com/products/individual#Downloads(到2021.4为止,内置Python=3.8)
清华镜像下载(含旧版,下载速度很快):
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
Anaconda与Python版本对照:
https://zhuanlan.zhihu.com/p/335319400
7. Anaconda安装(注意配置环境变量)
https://blog.csdn.net/Python_Smily/article/details/105993200
Pytorch安装(使用的终端是Anaconda Prompt)
8. 设置镜像,保障下载速度
conda config –add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config –set show_channel_urls yes
9.处理condarc
位置:Anaconda\pkgs\conda-4.6.11-py37_0\info\test\tests(进入pkgs后,可以直接搜索condarc)
处理:对红框部分,删去- defaults,把网址换成国内镜像,如:http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
- 搭建虚拟环境
https://www.bilibili.com/video/BV1Qk4y117nn?from=search&seid=3449121096884816059 - 安装Pytorch
官网:https://pytorch.org/
旧版本:https://pytorch.org/get-started/previous-versions/
把指令稍作修改,去掉-c,如下
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 -c pytorch
修改为:
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 pytorch
然后在Anaconda Prompt中运行该指令
11.安装显示done之后,import torch还是报错,加上下文中的步骤
https://blog.csdn.net/weixin_41608328/article/details/103986181
12.测试是否安装成功
ipython
import torch
print(torch.cuda.is_available())
输出为True,则成功;输出为ModuleNotFoundError: No module named ‘torch’,则失败。
安装了两个周,主要踩的坑是:
报错:failed with initial frozen solve. Retrying with flexible solve.
(网上许多解法是运行 conda config --set channel_priority flexible,但是没能解决我的问题,照此仍旧是报这个错误。后来是发现我原本安装的最新版的anaconda带的是Python3.8,另外安装的CUDA9.0不能兼容这么高版本的Python。所以该换了低版本的anaconda,然后走以上流程,最后成功安装。)
总结:Anaconda对应Python版本,显卡对应CUDA版本,CUDA和Python对应Pytorch版本。把三个对应关系都正确配置,其它的路都比较好走。
虽然费时挺长,中间更换了操作系统,安装了多个版本的Pytorch和Aanaconda,报了各种错误,经历几次心灰意冷,一度要配新的电脑,总算是成功了,很是开心。感谢这么多的大佬做分享,写得都很细节。不过我是因为没有搞清各部分的对应关系,还是走到歪道上去了。希望该文能把我踩的坑给补上,让后来者可以很快就安装成功,毕竟这只是一件小事情,风暴在后面呢。