首页 > 编程学习 > pytorch安装过程梳理及踩坑总结

pytorch安装过程梳理及踩坑总结

发布时间:2022/5/14 15:35:25

前言:前前后后安了两个周,所以暂时还没安装成功的朋友不要着急,再慢也慢不过我了。

大概流程(以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

  • 在这里插入图片描述
  1. 搭建虚拟环境
    https://www.bilibili.com/video/BV1Qk4y117nn?from=search&seid=3449121096884816059
  2. 安装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,报了各种错误,经历几次心灰意冷,一度要配新的电脑,总算是成功了,很是开心。感谢这么多的大佬做分享,写得都很细节。不过我是因为没有搞清各部分的对应关系,还是走到歪道上去了。希望该文能把我踩的坑给补上,让后来者可以很快就安装成功,毕竟这只是一件小事情,风暴在后面呢。

Copyright © 2010-2022 ngui.cc 版权所有 |关于我们| 联系方式| 豫B2-20100000