pytorch环境配置

1、首先下载安装anaconda

需勾选环境变量，若未勾选则需手动添加。

2、安装pytorch

打开anaconda prompt

若开头有括号（base）即安装成功

输入

conda create -n （虚拟环境的名称） python=x.x

创建虚拟环境，输入conda env list，若显示刚刚创建的虚拟环境则创建成功

输入

activate （虚拟环境名称）

进入虚拟环境， 若开头括号显示创建的虚拟环境名称则成功

 

cmd命令行输入nvidia-smi，可查看本电脑的显卡及版本

此处推荐一个安装办法（参考自windSeS博主(10条消息) Python安装Pytorch教程（图文详解）_pytorch安装_振华OPPO的博客-CSDN博客）：

在清华源中找到与自己cuda版本对应的pytorch版本

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/然后在虚拟环境下输入

conda install pytorch==1.7.0 torchvision==0.8 cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch

其中torchvision版本和cudatookkit版本参考下面链接或自行搜索(10条消息) PyTorch碎片：PyToch和Torchvision对应版本_pytorch和torchvision版本_柚有所思的博客-CSDN博客安装成功后输入python

import torch没有报错即安装成功，输入torch.cuda.is_available()输出true即成功安装gpu版本（若安装cpu版本的pytorch则会输出false）