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版本

四、安装Pytorch

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

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

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

 2.进入python后继续输入：

import torch
import torchvision
torch.cuda.is_available()

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