Pytorch：显卡驱动版本、Pytorch版本的关系

Pytorch：显卡驱动版本、Pytorch版本的关系

引用

CSDN

1.

https://blog.csdn.net/weixin_45791458/article/details/140505241

显卡驱动程序一般是显卡自带的，可以在终端中使用nvidia-smi命令查看，如下图所示。

图1 查看驱动程序的版本

图1显示，本机的驱动程序版本为472.19，支持的最新CUDA Toolkit版本为11.4，更详细的驱动程序和CUDA Toolkit的对应关系如图2所示。

图2 驱动程序和CUDA Toolkit的对应关系

CUDA拥有两类API，一种是驱动API(Driver API)，一类是运行时API(Runtime API)，驱动API(Driver API)在驱动程序安装后就已可用，而运行时API(Runtime API)则需要额外安装CUDA Toolkit才可用，它是对驱动API的封装。

如果想更新显卡驱动程序，首先可以在英伟达官网进行下载。另外，在安装CUDA Toolkit时，也有选项允许下载驱动程序，但此时安装的驱动程序主要是为了开发和测试用途设计的，不适合在生产环境中与Tesla GPU搭配使用。

安装较新版本的Pytorch

conda install pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-cuda=11.8 -c pytorch -c nvidia
pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu118  

无论使用哪条命令，都会同时下载依赖项cuda runtime包，里面包含了Pytorch需要的运行时API。可以在下面的路径（或类似）中找到运行时API。

/usr/local/lib/python3.10/dist-packages/nvidia/cuda_runtime/lib/libcudart.so.11.0  

安装较老版本的Pytorch

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch
pip install torch==1.12.1+cu102 torchvision==0.13.1+cu102 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu102  

其中如果使用conda安装，会同时下载cudatoolkit包（仅供Pytorch使用）；如果使用pip安装，会在Pytorch包的Lib中自带运行时API，如下所示。

/usr/local/lib/python3.10/dist-packages/torch/lib/libcudart-a7b20f20.so.11.0  

注意事项

需要注意的是，Pytorch附带安装的运行时API版本，应低于驱动所限制的CUDA Toolkit版本（因为运行时API是基于驱动API的），例如对于图1所示的驱动版本，CUDA 11.4版本以上的Pytorch就可能会出现问题。