深度学习-Pytorch环境安装

一、pytorch环境安装(windows)

1、下载安装Anaconda

• 下载Anaconda：Anaconda3-2021.11-Windows-x86.exe ，然后安装路径自定义如"D:\Programmer\Anaconda3"，然后手动添加环境变量，其余一路next
  
  

2、查看cuda version信息

• 查看自己电脑上显卡信息：打开cmd窗口输入nvidia-smi显示GPU当前的状态，可以看到本机cuda版本是11.2 ， 查看cuda与显卡驱动的映射表
  
  
  

3、开始安装pytorch

• ① 注意事项：如果conda执行下面步骤②时报错CondaHTTPError: HTTP 000 CONNECTION FAILED和 Solving environment: failed with，修改C盘用户目录下文件 .condarc为如下文件内容 ，以及在cmd输入以下命令尝试将conda降级；否则忽略本步骤

  ssl_verify: true
  show_channel_urls: true
  
  channels:
    - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
    - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
  

  conda config --set allow_conda_downgrades true
  conda install conda=4.6
  

• ② 正式开始安装pytorch：CUDA 各个版本对应安装命令搜索 ，由于本机显卡驱动是cuda11.2，官网没有11.2，所以这里选择了CUDA 11.1安装；然后打开cmd，用conda建虚拟环境，并下pytorch执行如下命令
  

  conda create --p=D:\Programmer\Captcha_env python=3.7
  activate D:\Programmer\Captcha_env
  conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
  

• ③ 以上没问题安装好后，新建项目并设置前面的虚拟python环境为当前项目环境，然后建一个测试文件测试下如下代码是否安装成功了，如果返回True则成功了。
  

  import torch
  print(torch.cuda.is_available())
  

• ④ 如果报错Original error was: DLL load failed: 找不到指定的模块。虚拟环境cmd窗口执行pip uninstall numpy 和pip install numpy