【PyTorch】Ubuntu20.04下CUDA及GPU版本PyTorch的安装

Ubuntu20.04下CUDA及GPU版本PyTorch的安装

本文包含：

1. NVIDIA 驱动安装过程
2. CUDA 安装过程
3. GPU 版本的PyTorch安装

1 驱动安装

Ubuntu20.4 卸载老旧的NVIDIA驱动_ubuntu卸载nvidia显卡驱动_Coder802的博客-CSDN博客

# 安装
# 在官网下载对应驱动 https://www.nvidia.cn/geforce/drivers/

# 卸载
sudo ./NVIDIA-Linux-x86_64-495.46.run --uninstall
# 或者
sudo nvidia-uninstall

# 查看NVIDIA驱动版本及最大支持的CUDA版本
# 最大支持版本可向下兼容 即实际安装版本可低于最大版本
nvidia-smi

2 CUDA安装

1. 首先在PyTorch官网 找到安装pytorch需要的CUDA版本

• 11.7版本的下载页面

2. 下载 run file

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

• run file下载中

• 安装

• 因为前面已经安装了NVIDIA驱动，那么此处要取消勾选Driver

3. 安装成功

3 PyTorch安装

# 首先创建并激活你想要的python虚拟环境
conda create --name py38 python=3.8 -y
conda activate py38

# 官方给的命令 - 安装pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

• 可以看到下载的都是GPU版本的包

• 最后 - 验证安装 - 打印出版本号即说明安装成功

python -c "import torch; print (torch.__version__)"