ubuntu22.04 搭建 Pytorch环境

关于电脑

第一步 安装anaconda

1. 进入官网

链接: anaconda

2. 下载linux的sh版

3. 在对应位置输入

sh 文件名.sh

4. 选择 yes

5. 选择no

（我们需要自己配置环境）

安装完成anconda，接下来将要配置环境

第二步 完成安装后配置环境

1. 编辑环境变量

gedit ~/.bashrc

2. 添加内容

export PATH="/自己对应的路径名/anaconda3/bin:$PATH"

完成后如下图：

3.激活修改的内容

source ~/.bashrc

4. 测试

conda

若没有显示not fond 则表示安装成功

第三步 创建自己的虚拟环境

安装完成后，在当前终端输入：

source activate

即可进入base环境，这是我们使用anaconda安装环境后，本地默认环境

1. 创建自己的环境

环境名称为pytorch，可以替换成自己的名称

conda create -n pytorch

2. 进入刚刚创建的pytorch环境

conda activate pytorch

第四步在base环境下安装cuda

1. 安装 nvidia-cuda-toolkit 工具

sudo apt-get install nvidia-cuda-toolkit

2. 检查系统推荐显卡驱动，记录recommend选项

sudo ubuntu-drivers devices

3.添加驱动源

我是之前已经安装好了驱动（虽然不是recommend）所以没有这一步也没问题。

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt-get update

4.在 软件和更新 选择 附加驱动

更新完成后按要求重启

5. 测试

nvidia-smi

6. 下载cuda

cuda官网
这边我下载的是cuda11.7（按照nvidia-smi推荐下载）

接下来按照如下选择去下载
（ubuntu 版本一定要正确）

7.安装cuda

按照下方给出的命令下载安装

wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run

sudo sh cuda_11.7.0_515.43.04_linux.run

注意在选择时不要勾选安装驱动

8. 环境配置

sudo gedit ~/.bashrc

在末尾加入如下内容
注意将路径替换为自己的路径

export PATH="/自己对应的路径名/anaconda3/bin:$PATH"
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/extras/CPUTI/lib64
export CUDA_HOME=/usr/local/cuda/bin
export PATH=$PATH:$LD_LIBRARY_PATH:$CUDA_HOME

9. 测试

source ~/.bashrc

nvcc -V

表示安装成功

第五步 在自己创建的虚拟环境中安装pytorch

1.进入官网

pytorch

2.终端进入自己的虚拟环境

source activate

conda activate pytorch

3.选择对应版本并安装

选择pytorch版本
选择好对应的系统
选择使用什么安装，conda里可以使用pip安装的
选择语言：python
选择cuda版本：我的版本是11.7没有这个选项，11.6也可以使用

4.在终端输入官网复制的命令

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

我这里已经是安装完成，
至此，pytorch的安装已经完成