pytorch 支持amd显卡吗_2020-06-12 ubuntu系统下，pytorch安装

ubuntu系统下，pytorch安装

因为pytorch依赖于cuda的版本，显卡驱动的版本，三者之间对应不上就会安装失败

1.显卡驱动下载

显卡，支持最新的驱动型号

TitanX430.64 LINUX X64 (AMD64/EM64T) DISPLAY DRIVER

TitanXP440.82 LINUX X64 (AMD64/EM64T) DISPLAY DRIVER

GTX 1080Ti440.82 LINUX X64 (AMD64/EM64T) DISPLAY DRIVER

RTX 2080Ti440.82 LINUX X64 (AMD64/EM64T) DISPLAY DRIVER

2.驱动型号查询

nvidia-smi

3.驱动型号对应支持的cuda的版本号

[来源]https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

CUDA ToolkitLinux x86_64 Driver VersionWindows x86_64 Driver Version

CUDA 11.0.189 RC>= 450.36.06>= 451.22

CUDA 10.2.89>= 440.33>= 441.22

CUDA 10.1 (10.1.105 general release, and updates)>= 418.39>= 418.96

CUDA 10.0.130>= 410.48>= 411.31

CUDA 9.2 (9.2.148 Update 1)>= 396.37>= 398.26

CUDA 9.2