作者:非妃是公主
专栏:《python学习》
个性签:顺境不惰,逆境不馁,以心制境,万事可成。——曾国藩
转载请标明,原文链接:https://blog.csdn.net/myf_666/article/details/122706716
1、创建虚拟环境
打开Anaconda prompt
// 输入命令,如下图
conda create -n pytorch python=3.8
等待虚拟环境配置完成!
2、查看虚拟环境,激活虚拟环境
//查看虚拟环境
conda env list
//激活虚拟环境
conda activate pytorch
3、安装pytorch,打开pytorch官网.
网站已经帮助我们检测显卡型号,配置好相应的安装版本:
//复制红圈中的指令:比如我自己的如下
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
安装界面如下:
等待安装完毕即可
4、检验环境是否配置成功:
在anaconda prompt中输入
python
import torch
torch.cuda.is_available()
如果出现True
,则表示环境配置成功!