Conda配置pytorch

目录

一、conda创建虚拟环境(环境名称为FCN,可随意更改)

二、激活虚拟环境

三、查看cuda版本

四、安装Pytorch

五、测试pytorch是否安装成功

一、conda创建虚拟环境(环境名称为FCN,可随意更改)

conda create -n FCN python==3.8

二、激活虚拟环境

刚刚创建了FCN的虚拟环境,之后的操作均在FCN下进行

在命令行输入如下代码进入环境:

conda activate FCN

三、查看cuda版本

1.在命令行输入nvidia-smi,红框中展示的为你的cuda版本

Conda配置pytorch

四、安装Pytorch

1.查看与自己电脑系统对应的pytorch版本

pytorch官网(https://pytorch.org/get-started/previous-versions/)可以查看不同pytorch版本在不同操作系统的安装方式。以下为示例(可以根据自己的系统和cuda版本进行选择):

Conda配置pytorch

2.我的cuda版本是11.4,系统是Lunix

目前最新的pytorch版本是1.12.0,我选择安装pytorch版本为1.11.0:

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

如果显卡不支持CUDA,也可以安装CPU版本的pytorch,例如使用:

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cpuonly -c pytorch

五、测试pytorch是否安装成功

1.在FCN环境下进入python,输入:

python

Conda配置pytorch

2.进入python后继续输入:

import torch
import torchvision
torch.cuda.is_available()

3.如果看到如下则证明安装成功:

Conda配置pytorch

Original: https://blog.csdn.net/baidu_38662087/article/details/127504440
Author: 小薛薛snow
Title: Conda配置pytorch

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/770584/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球