发布时间:2023-04-20 文章分类:电脑百科 投稿人:王小丽 字号: 默认 | | 超大 打印

作者:非妃是公主
专栏:《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,则表示环境配置成功!