win11+cuda+cudnn+pytorch安装

文章目录

  • 一、前言
  • 二、安装软件
    • 1 anaconda和pycharm
    • 2 CUDA+cudnn+pytorch
  • 三、总结

一、前言

我的电脑是,联想拯救者y9000p2022,win11系统3060显卡。这两天要跑一个深度学习的项目,开始着手软件的安装和环境的安装,期间看来不少教程,踩了不少坑,这里记录一下防坑要点、好用的教程,以便以后查找,也给大家分享下,少浪费点时间。如果是第一次装的话,建议大家先把这篇文章先浏览一遍。

二、安装软件

1 anaconda和pycharm

anaconda安装比较简单,直接按照这里的教程就行anaconda和pycharm安装教程,这里我说下要点。安装的时候,一定要点install for just me
win11+cuda+cudnn+pytorch安装_第1张图片
要不然下一步就无法自动添加环境变量,需要手动添加
win11+cuda+cudnn+pytorch安装_第2张图片

我没有手动添加过,大家这里不能点的话,只能手动添加了,手动添加环境变量,安装成功后记得用命令测试下,确定没有问题。

  • anaconda可以创建出多个虚拟的环境,每个环境中可以配置不同版本的python,有时候不同的项目需要的python版本不同,就需要用conda创建相应的环境了。对于已经安装python的电脑,也没有关系,不冲突。
  • anaconda大部分都没安装到c盘,当你创建python环境时,conda会把相关文件放到C:\Users\用户名.conda中,很占内存。我们需要修改默认C盘存放路径。
  • 在conda创建的环境中能跑出代码,就说明这两个软件安装好了,后面安装的软件出现问题,就不用管这两个了。

2 CUDA+cudnn+pytorch

cuda安装本身没有多大难度,关键就是电脑自身驱动版本和安装的CUDA版本,pytorch版本必须要相匹配。大家想了解的话,可以看下这个CUDA版本,我的电脑是这样的
win11+cuda+cudnn+pytorch安装_第3张图片
我之前一直瞎装,全部都往最高版本装,结果多次安装都失败。

  • 重点!重点!重点! 亲测有效安装版本,安装版本推荐,一定要按照该博主的版本安装
  • 按照博主的cuDNN v8.1.1 (Feburary 26th, 2021), for CUDA 11.0,11.1 and 11.2​下载后的文件名为cudnn-11.2-windows-x64-v8.1.1.33.zip,不要怀疑这个11.2而CUDA版本为11.1,没问题的。
  • 有的CUDA安装教程需要添加环境变量,大家安装好后先检验是否安装成功,成功了就不需要手动添加环境变量了。安装这几个软件我都没有手动添加过环境变量。
  • CUDA一定要选择自定义安装,方便记住文件位置
  • 我的电脑已经了安装vs,安装出问题的话,大家可以考虑一下点
  • CUDA安装的各个教程都大同小异,大家结合着看,CUDA安装教程,CUDA安装,这里面包含了CUDA和pytorch的安装,只是大家的版本不同
  • 检查CUDA和pytorch是否安装成功,安装成功检验,要先检验CUDA安装是否成功,再安装pytorch
  • 把安装命令的-c pytorch去掉可以明显提高下载速度

pytorch安装失败,有些教程会配置镜像文件路径,方便快速下载,但可能会出点问题,比如:CondaHTTPError: HTTP 000 CONNECTION FAILED for url或者CondaHTTPError: HTTP 403 FORBIDDEN for url,解决方法,但我试过后没有用,可能每个人的电脑有差别吧。之后我还是用了原命令,没用用国内的镜像

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

下载速度慢了很多(一两个小时),最后只有一个libtiff-4.0.2-1.tar.bz2文件没安装好,报错。要记录下有哪些文件没有安装成功,自己单独去下载包
win11+cuda+cudnn+pytorch安装_第4张图片
之后,我将pkgs文件夹下的这一类文件夹删除
win11+cuda+cudnn+pytorch安装_第5张图片
这其实就是安装pytorch过程中下载解压的文件,但要保留它们的压缩文件(同样在pkgs文件夹中)。
win11+cuda+cudnn+pytorch安装_第6张图片
保留bz2文件,以防万一,将它们放到另一个文件夹中,我把它们放到了桌面的000-副本文件中
win11+cuda+cudnn+pytorch安装_第7张图片
之后在该文件夹中打开cmd,输入命令安装(conda install 安装包),它们还是会安装到pkgs文件夹中。之前没有安装好的文件,也要另外去找安装包。30多个安装包,单个安装,哪个有问题就解决哪个,然后在python中import torch,没有报错,成功!

conda install blas-2.116-mkl.tar.bz2

没有安装成功的libtiff-4.0.2-1.tar.bz2包,直接搜,libtiff地址当下载完libtiff-4.0.2-1.tar.bz2包安装时,仍报错,大概意思是无法在某个网站中找到该包,下载另一个版本libtiff-4.0.6-vc14_9.tar.bz2成功安装。
win11+cuda+cudnn+pytorch安装_第8张图片

  • 在使用yolo5时,pip install -r requirements.txt会有警告,解决方法
    在这里插入图片描述
    但重启计算机,结果是pycharm打不开了,好像和破解插件有关。将其设置关闭后,pycharm能够打开,也能跑程序,这个警告无伤大雅。

三、总结

从开始装软件到第一个程序跑通,断断续续花了三天,最重要的还是CUDA和pytorch的版本问题。我的这篇总结肯定不能把所有的问题都记录下来,大家多对比着看几篇,希望大家也能够快点把软件装好。

你可能感兴趣的:(笔记,pytorch,python,深度学习)