Ubuntu20.04配置Pytorch环境（GPU版）

1.安装nvida驱动

打开终端执行：ubuntu-drivers devices

sudo apt install nvidia-driver-535  #安装535驱动

2.查看显卡驱动版本

通过终端 nvidia-smi  查看

3.根据对应版本安装CUDA

CUDA Toolkit 11.8 Downloads | NVIDIA Developer

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/

cuda_11.8.0_520.61.05_linux.runsudo sh cuda_11.8.0_520.61.05_linux.run

添加环境变量

export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

 4.安装pytorch

Start Locally | PyTorch

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