【环境配置】新建一个pytorch-gpu的虚拟环境

我的Cuda版本： 11.2

Python：3.9.6

前提：你电脑已经有nividia驱动程序，cuda，cudnn

一、用anaconda建立一个新的虚拟环境

指令： conda create --name xxx python=3.8 -y 

eg:  conda create --name mylab python=3.8 -y  【mylab是我虚拟环境的名称】

创建成功！

正常来说创建的新虚拟环境会放在anaconda下的envs下，但是我不知道为什么我每次新建都会自动放在了MyTools下，跟anaconda是同级的关系。因为不影响使用，我也没有深究，我想应该是以前可能无意间设置过什么默认设置吧。

二、安装pytorch

1. 激活环境

conda activate xxx  

eg: activate mylab  【我的电脑是需要把conda去掉，才成功运行和激活】

你看前面括号带个路径就是说你成功激活了虚拟空间

2. 下载对应版本pytorch【我想要gpu版本的pytorch】

激活环境后，在终端输入在终端输入指令（pytorch-cuda指令就会下载对应版本的gpu-pytorch，之前不知道在哪体现它是gpu还纠结了老半天）：

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

安装完GPU版本后，你可以使用torch.cuda.is_available()来检查PyTorch是否成功安装并支持GPU。如果返回True，则表示已成功安装GPU版本并可以在GPU上运行PyTorch代码。

返回True代表终于成功啦！