cudn+cudnn+pytorch安装,环境配置

文章目录

    • 0. 前提:
    • 1. cudn、cudnn下载
    • 2. pytorch下载、环境配置:
    • 3.出现的错误

0. 前提:

本人之前已经安装过anaconda+python+pycharm。学习中发现要用到GPU
anaconda:3.9
conda版本:conda 4.12.0(Anaconda3文件夹–>在anaconda prompt中输入conda -V)
在这里插入图片描述


1. cudn、cudnn下载

参考:CUDA、CUDNN在windows下的安装及配置


2. pytorch下载、环境配置:

  1. 打开Anaconda prompt 窗口,如下图所示:
    cudn+cudnn+pytorch安装,环境配置_第1张图片
  2. 输入 conda create –n pytorch python=3.8.1
    (关于python版本:按 WIN +R 进入运行,输入 cmd,点击确定打开命令提示符,输入python -V,进入 python 交互解释器,即可显示 python 版本信息)
    在这里插入图片描述
  3. 提示Remove existing environment:输入y移除环境
    cudn+cudnn+pytorch安装,环境配置_第2张图片
  4. 提示Proceed:输入y安装
  5. 需要激活环境,为后面在这个环境内安装pytorch包做准备,激活命令conda activate pytorch
    输入指令pip list,可以看看我们这个环境里面有哪些安装包。
    cudn+cudnn+pytorch安装,环境配置_第3张图片
    3.安装pytorch包

1、查看任务管理器–>性能—>是否有显卡GPU(判断安装pytorch时是GPU/CPU)cudn+cudnn+pytorch安装,环境配置_第4张图片

2、win+r–>cmd–>输入nvidia-smi—>查看CUDA版本(我的是11.1)
cudn+cudnn+pytorch安装,环境配置_第5张图片

3、登陆pytorch的官网,下滑至如下界面,按照自己的系统选择,关于选择GPU还是CPU如上。
参考李沐:Pytorch CUDA安装视频
cudn+cudnn+pytorch安装,环境配置_第6张图片
由于我的CUDA最高能支持11.1,找以前的版本
cudn+cudnn+pytorch安装,环境配置_第7张图片

4、输入获取的指令
输入pip --no-cache-dir --default-timeout=1000 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
其中 --no-cache-dir:不使用缓存。
--default-timeout=1000 :防止出现read time out问题,下载速度变慢,pip下载超时。cudn+cudnn+pytorch安装,环境配置_第8张图片

5、检查是否安装成功。
输入python---->输入import torch (if不报错)---->输入torch.cuda.is_available(),返回True。
至此,pytorch安装完成。( •̀ ω •́ )✧
cudn+cudnn+pytorch安装,环境配置_第9张图片


3.出现的错误

  1. 下载pytorch不是用的官方给出的pip指令,用了清华的镜像。torch.cuda.is_available()报错!!!
    原因:Nefu_lyh,被conda镜像安装给坑了。以为下载的是GPU版本,其实镜像下载的cpu版本。
    winr+r—>cmd—>python---->import torch---->print(torch.__version__)---->清华镜像下载的不是GPU。
  2. 所以 必须去官网用pip下载安装。
    cudn+cudnn+pytorch安装,环境配置_第10张图片

参考每天进步一点点

你可能感兴趣的:(安装-配置,python,pycharm,jupyter)