我的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代表终于成功啦!