【AI】Ubuntu20.04安装cuda、cuDNN、Anacodna、PyTorch

1、版本选择

1)选择PyTorch版本:2.0.1
【AI】Ubuntu20.04安装cuda、cuDNN、Anacodna、PyTorch_第1张图片

2)对应的CUDA版本:11.7或者11.8,选择较新的11.8
3)确定显卡是否支持,使用命令nvidia-smi查看显卡驱动版本及支持的CUDA最高的版本,本人的为12

【AI】Ubuntu20.04安装cuda、cuDNN、Anacodna、PyTorch_第2张图片
4)cuDNN版本
选择和CUDA版本对应的最新版本v8.9.1
【AI】Ubuntu20.04安装cuda、cuDNN、Anacodna、PyTorch_第3张图片

2、下载

2.1 下载CUDA

各个版本CUDA下载

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

本人选择11.8
【AI】Ubuntu20.04安装cuda、cuDNN、Anacodna、PyTorch_第4张图片
选择对应的系统版本
【AI】Ubuntu20.04安装cuda、cuDNN、Anacodna、PyTorch_第5张图片在终端中使用命令下载

你可能感兴趣的:(AI,人工智能,pytorch,深度学习)