鉴于很多论文用了比较老的 Python、PyTorch 以及 CUDA 环境,基于 4090D 复现代码时难免遇到很多困难,这里做一个记录,便于以后翻阅。

Python 版本问题

Python 的高版本如 Python 39 对于低版本如 Python 36、37、38 基本上都是可以兼容的,有问题的地方在于有一些老的功能被移除了,其次就是其他的一些包的低版本可能依赖一些低版本的 Python。
如果在复现论文时,因为一些包的版本问题需要抬高 Python 版本,一般只需要注意其他包的适配问题,Python 本身一般不会出现很大问题。

PyTorch 和 CUDA 版本问题

PyTorch 最常见的问题可能是论文中使用的 PyTorch 版本对应的 CUDA 低于当前 GPU 所能运行的最低 CUDA。
4090D 可以使用的 CUDA 最高版本可以使用以下命令查看:

1
nvidia-smi

下图显示的是 CUDA 支持的最高版本:
查看结果
但通常情况下,4090D 还有一个支持的最低版本:

1
2
python=3.9
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia
1