第一步配置镜像源安装必要环境

我在这篇文章中详细写了如何配置清华源
更换镜像

第二步安装anaconda

1 进入官网

https://www.anaconda.com/

2 下载linux的sh版

3 在对应位置通过命令输入

sh 文件名.sh

4 选择 YES

5 选择 NO 我们要自定义配置环境

第三步安装完成Anconda后配置环境

1 编辑环境变量

vim ~/.bashrc

2 添加内容

这里的自己对应的路径名一般默认安装会有提示一般都是home/用户名

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

3 激活修改的内容

source ~/.bashrc

4 测试

conda

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

第四步通过Anaconda创建pytorch空间用于pytorch

前提知要

使用anaconda安装环境后本地默认环境成为base环境

自己创建的环境通过 conda activate 空间名字环境名字进入

退出自己创建的环境到默认环境conda deactivate

1 创建空间,命名为pytorch

conda create -n pytorch

2 进入pytorch空间

conda activate pytorch

ps若在执行命令的时候出现一下问题

执行

source activate

第五步在base环境中 安装cuda

1 安装 nvidia-cuda-toolkit 工具

sudo apt-get install nvidia-cuda-toolkit

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

sudo ubuntu-drivers devices

2022.12.14更新记录记录下recommend前面的数字比如这里470到时候更新驱动选择该数字短的驱动

3 添加驱动源

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

sudo apt-get update

4 在Ubuntu系统中找到 软件和更新 选择 驱动有更新记录看下面文字

2022.12.14更新记录这里不要选择第一个可能会出现黑屏现象。这里选择其它同名的。

等待更新完毕后再reboot

reboot

5 重启后的测试

nvidia-smi

记录 cuda 版本

6 下载cuda

官网

https://developer.nvidia.com/cuda-toolkit-archive

我的是11.4

选择11.4的哪个版本都可以接下去按照如下选择去下载

注意linux的版本要确保正确

7 安装cuda(执行自己上面红框的内容)

wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.runsudo

2022.12.14更新记录如果sh无反应或者需要权限

1. chmod 755 <sh文件>
2. sudo sh <sh文件>

sh cuda_11.4.4_470.82.01_linux.run

2022.12.14更新记录这里选择是Continue截图的时候有问题

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

cd ~/NVIDIA_CUDA-11.4_Samples/1_Utilities/bandwidthTest/

make

./bandwidthTest

第六步在pytorch环境中 安装 pytorch环境

1 进入官网

https://pytorch.org/

2 安装

选择pytorch版本
选择好对应的系统
选择使用什么安装conda里可以使用pip安装的
选择语言python
选择cuda版本我的版本是11.4

复制命令

3 通过命令进入pytorch环境

source activate

conda activate pytorch

4 执行自己官网复制的命令

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

done撒花

番外

删除conda环境

conda uninstall -n 名称 --all