一、下载及安装Anaconda
打开官网进行下载,本人为Windows,点击下载Anaconda,如果想下载其他版本,点击网址。
下载到本地点击安装
过程如下,逐步进行
安装完成后,按Win键找到Anaconda3文件夹,打开点击Anaconda Prompt。
(base)为自带环境,可输入以下命令查看存在的环境
conda env list
创建自己的虚拟环境conda create -n 环境名字(英文) python=x.x(python版本),如下,创建一个名字叫DL,python为3.8版本的环境。
输入命令激活创建的环境,会发现环境名发生变化,执行conda env list命令也会发现多了一个环境名。
conda activate 环境名
二、安装pytorch
安装pytorch,pytorch的官网在国外,下载相关的包速度较慢,给环境换源,在新创建的环境下执行以下的命令给环境换成清华源,复制粘贴即可。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
下载pytorch前,查看自己电脑显卡以及支持的cuda版本,按Win+R,输入cmd打开窗口,输入以下命令查看,红框内为支持的cuda最高版本,建议使用低于最高版本的cuda,本人使用电脑能支持的最高版本报错,自行斟酌。
nvidia-smi
打开网址,可在当前也选择自己适配的版本,也可点击左下角红框内选择其他版本。
本人选择pytorch1.10.1,cuda为10.2版本,不要输入结尾的 -c pytorch,不然仍从国外源下载,输入命令运行即可。
三、使用pycharm验证环境是否安装成功
打开pycharm,点击右下角,如图所示。
点击添加解释器
按照下图步骤依次执行,添加环境,不同版本pycharm界面有所不同。
验证环境:
创建.py文件,将下列代码添加进去执行,验证是否成功。
import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())
如下图所示即为成功:True表示安装成功,另外另个表示cuda和cudnn的版本。