目录
参考:
ubuntu22.04 搭建 Pytorch环境_ubuntu pytorch_不是我吧12138的博客-CSDN博客
1.安装Anaconda
step1:下载anaconda安装包
打开官网链接 Anaconda | The World’s Most Popular Data Science Platform
选择 Linux 下载。
step2:执行安装命令
打开终端
sh Anaconda3-2023.03-1-Linux-x86_64.sh
选择yes,yes
完成anaconda安装。
step3:配置环境变量
打开终端,配置环境变量:
sudo gedit ~/.bashrc
在配置文件最后一行添加:(注意路径要替换为对应的自己的路径)
export PATH="/home/wcx/anaconda3/bin:$PATH"
step4:测试
打开终端
conda
表示安装成功。
2.在虚拟环境下安装Pytorch
step1:anaconda创建虚拟环境
创建虚拟环境
conda create -n ORFD python==3.8
激活环境
source activate ORFD
ps:删除虚拟环境:
conda remove -n ORFD --all
ps:显示环境列表
conda env list
带星号* 表示当前激活的环境。
step2:在官网选择对应版本Pytorch
打开官网 PyTorch
选择对应版本后。
step3:安装Pytorch
打开终端,输入命令官网的命令
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
step4:测试pytorch及版本
确保在安装pytorch的环境下,输入
python
依次输入
>>> import torch
>>> print(torch.version.cuda)
安装完成!