一、anaconda

1.1下载anaconda

anconda安装 pytorch anaconda下载pytorch_pytorch

 1.2安装anaconda

anconda安装 pytorch anaconda下载pytorch_pytorch_02

 

anconda安装 pytorch anaconda下载pytorch_pytorch_03

 

anconda安装 pytorch anaconda下载pytorch_pycharm_04

 

anconda安装 pytorch anaconda下载pytorch_anconda安装 pytorch_05

 这里选择安装的位置,建议放C盘

anconda安装 pytorch anaconda下载pytorch_python_06

点install,完成安装 

1.3配置anaconda环境变量

找到刚才安装的目录路径,在系统变量path里添加如下图三个环境变量

anconda安装 pytorch anaconda下载pytorch_anconda安装 pytorch_07

anconda安装 pytorch anaconda下载pytorch_pycharm_08

 1.4测试anaconda

1.4.1win+r一起按呼出运行,输入cmd并回车

anconda安装 pytorch anaconda下载pytorch_pycharm_09

1.4.2输入如下代码并回车

conda --version

 

anconda安装 pytorch anaconda下载pytorch_anconda安装 pytorch_10

出现上图版本信息则安装成功。 

二、cuda

2.1查看自己电脑显卡对应的cuda版本

2.1.1桌面-》点击鼠标右键-》NVIDIA控制面板

anconda安装 pytorch anaconda下载pytorch_虚拟环境_11

 2.1.2点击系统信息

anconda安装 pytorch anaconda下载pytorch_虚拟环境_12

 2.1.3点击组件

anconda安装 pytorch anaconda下载pytorch_pytorch_13

上图NVCUDA64.DLL后对应的产品名称 11.8.87则是最高可运行的版本

2.2下载cuda

2.2.1下载cuda

2.3安装cuda

anconda安装 pytorch anaconda下载pytorch_anconda安装 pytorch_14

安装过程中全部选择默认,安装方式选择“精简安装”即可

2.4检测cuda是否安装成功

 2.4.1win+r一起按呼出运行,输入cmd并回车

anconda安装 pytorch anaconda下载pytorch_python_15

2.4.2输入以下代码并回车

nvcc -V

 

anconda安装 pytorch anaconda下载pytorch_pycharm_16

 出现上图显示即表示安装成功

三、pytorch

3.1查看anaconda里python版本

在开始菜单里,以管理员身份打开Anaconda prompt,输入如下代码:

python

anconda安装 pytorch anaconda下载pytorch_python_17

3.2更改anaconda 配置

这一步是为了下载更快,所以我们需要进行channels更改,第一次安装的anaconda是没有.condarc文件的,所以我们要在anaconda prompt里运行如下代码:

conda config

 运行完后会生成.condarc文件,建议大家下载everything直接搜索,将找到的.condarc文件用记事本打开,将如下文本复制粘贴替换进去:

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - defaults
  - conda-forge
show_channel_urls: true
default_channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
custom_channels:
  conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

3.3创建虚拟环境

在开始菜单里,管理员身份打开Anaconda prompt,输入如下代码:

conda create -n pytorch-gpu python=3.9

其中pytorch-gpu为所创建的虚拟环境名称,python=3.9为指定python版本,这里建议大家根据anaconda里的python版本进行创建。

3.4切换到虚拟环境

在开始菜单里,管理员身份打开Anaconda prompt,输入如下代码:

conda activate pytorch-gpu

切换到虚拟环境后,会显示如下界面:

anconda安装 pytorch anaconda下载pytorch_python_18

3.5下载安装pytorch

进入pytorch官网,下拉找到如下图界面

anconda安装 pytorch anaconda下载pytorch_pytorch_19

 根据自己的需要选择,注意一点:这里的cuda版本不能高于电脑本身的显卡cuda版本,详见2.1

复制上图里run this command里的代码,复制到3.4里切换到虚拟环境后的anaconda prompt里,回车。注意:这里一定要以管理员身份模式打开,不然有可能会报错。

等待.....中间会让你确认一次(输入y或n)y表示确定,n表示取消,输入y回车即可

然后就是等待它慢慢下载自行安装了。

3.6测试pytorch是否安装成功

在3.5小节安装好了后的anaconda prompt里输入以下代码:

python

anconda安装 pytorch anaconda下载pytorch_pycharm_20

接着运行以下代码:

import torch
torch.cuda.is_available()

 

anconda安装 pytorch anaconda下载pytorch_虚拟环境_21

显示为ture,则安装成功

四、在pycharm里配置 pytorch

4.1打开pycharm

新建一个项目:

anconda安装 pytorch anaconda下载pytorch_pytorch_22

选择我们已有的虚拟环境

在main.py里输入如下代码并运行:

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

 

anconda安装 pytorch anaconda下载pytorch_虚拟环境_23

 出现结果为ture则配置成功。

注意,这里有的同学可能会有点问题,如果出现问题,进入设置界面:

anconda安装 pytorch anaconda下载pytorch_pycharm_24

 这样设置好以后,点击apply和ok即可。