目录
一、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.如果看到如下则证明安装成功:
Original: https://blog.csdn.net/baidu_38662087/article/details/127504440
Author: 小薛薛snow
Title: Conda配置pytorch
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/770584/
转载文章受原作者版权保护。转载请注明原作者出处!