作者:非妃是公主
专栏:《python学习》
个性签:顺境不惰,逆境不馁,以心制境,万事可成。——曾国藩

python配置虚拟环境,并在虚拟环境中安装pytorch教程

转载请标明,原文链接:https://blog.csdn.net/myf_666/article/details/122706716
1、创建虚拟环境
打开Anaconda prompt
python配置虚拟环境,并在虚拟环境中安装pytorch教程

// 输入命令,如下图
conda create -n pytorch python=3.8

python配置虚拟环境,并在虚拟环境中安装pytorch教程
等待虚拟环境配置完成!

2、查看虚拟环境,激活虚拟环境

//查看虚拟环境
conda env list
//激活虚拟环境
conda activate pytorch

python配置虚拟环境,并在虚拟环境中安装pytorch教程
3、安装pytorch,打开pytorch官网.
python配置虚拟环境,并在虚拟环境中安装pytorch教程
网站已经帮助我们检测显卡型号,配置好相应的安装版本:

//复制红圈中的指令:比如我自己的如下
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

安装界面如下:
python配置虚拟环境,并在虚拟环境中安装pytorch教程
等待安装完毕即可
4、检验环境是否配置成功:
python配置虚拟环境,并在虚拟环境中安装pytorch教程
在anaconda prompt中输入

python
import torch
torch.cuda.is_available()

如果出现True,则表示环境配置成功!

声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。