一、anaconda
1.1下载anaconda
1.2安装anaconda
这里选择安装的位置,建议放C盘
点install,完成安装
1.3配置anaconda环境变量
找到刚才安装的目录路径,在系统变量path里添加如下图三个环境变量
1.4测试anaconda
1.4.1win+r一起按呼出运行,输入cmd并回车
1.4.2输入如下代码并回车
conda --version
出现上图版本信息则安装成功。
二、cuda
2.1查看自己电脑显卡对应的cuda版本
2.1.1桌面-》点击鼠标右键-》NVIDIA控制面板
2.1.2点击系统信息
2.1.3点击组件
上图NVCUDA64.DLL后对应的产品名称 11.8.87则是最高可运行的版本
2.2下载cuda
2.2.1下载cuda
网址1
如果网址1没有你可下载的版本,点击网址2
网址2
2.3安装cuda
安装过程中全部选择默认,安装方式选择“精简安装”即可
2.4检测cuda是否安装成功
2.4.1win+r一起按呼出运行,输入cmd并回车
2.4.2输入以下代码并回车
nvcc -V
出现上图显示即表示安装成功
三、pytorch
3.1查看anaconda里python版本
在开始菜单里,以管理员身份打开Anaconda prompt,输入如下代码:
python
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
切换到虚拟环境后,会显示如下界面:
3.5下载安装pytorch
进入pytorch官网,下拉找到如下图界面
根据自己的需要选择,注意一点:这里的cuda版本不能高于电脑本身的显卡cuda版本,详见2.1
复制上图里run this command里的代码,复制到3.4里切换到虚拟环境后的anaconda prompt里,回车。注意:这里一定要以管理员身份模式打开,不然有可能会报错。
等待.....中间会让你确认一次(输入y或n)y表示确定,n表示取消,输入y回车即可
然后就是等待它慢慢下载自行安装了。
3.6测试pytorch是否安装成功
在3.5小节安装好了后的anaconda prompt里输入以下代码:
python
接着运行以下代码:
import torch
torch.cuda.is_available()
显示为ture,则安装成功
四、在pycharm里配置 pytorch
4.1打开pycharm
新建一个项目:
选择我们已有的虚拟环境
在main.py里输入如下代码并运行:
import torch
print(torch.cuda.is_available())
出现结果为ture则配置成功。
注意,这里有的同学可能会有点问题,如果出现问题,进入设置界面:
这样设置好以后,点击apply和ok即可。