4090D 跑论文代码注意事项
鉴于很多论文用了比较老的 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 | python=3.9 |
1 |
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来源 Excelius's World!