关于使用pytorch在30系列显卡（高级别显卡）算力不够问题

问题

在后台搭建docker镜像环境时，在10、20系列卡上兼容常见版本的pytorch。搭建环境时在dockerhub、nvidia官网随便拉取镜像基本都能使用。但把镜像迁移到30系列卡时就会有算力不够的情况，主要原因就是30系列卡的算力升级了。。。详情见这个博客

具体日志

/opt/conda/lib/python3.7/site-packages/torch/cuda/init.py:125:
UserWarning:
NVIDIA GeForce RTX 3080 Ti with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 3080 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

意思是：NVIDIA GeForce RTX 3080 Ti与CUDA能力sm_86不兼容目前的PyTorch安装。
当前的PyTorch安装支持CUDA能力sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37。
也就是安装的cuda版本低了，但pytorch安装是要和cuda对应的，所以也能说是pytoch安装版本不对。。。

解决方法

• 升级torch版本
• 升级torch对应cuda版本

最后

以上就是称心豆芽为你收集整理的关于使用pytorch在30系列显卡（高级别显卡）算力不够问题的全部内容，希望文章能够帮你解决关于使用pytorch在30系列显卡（高级别显卡）算力不够问题所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。