本文主要是介绍RTX3080ti Pytorch踩坑指南,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
RTX3080ti Pytorch踩坑指南
**
2022.5.2日更新在最后,建议先看
**
- 写在前面的罗里吧嗦
都说3080不支持cuda10, 可是复现前两年的代码,对应低版本的pytorch,可怎么跑呢?
经过亲身试验,安装cuda10,跑代码各种cuda相关报错。
比如
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb
(解决方法:新建环境,pytorch官网pip安装。在最后)
最终找到了一条玄学之法。(仅个人经验)
如果在cuda110的上直接使用下面的命令装pytorch1.4.0,行不通。(来自pytorch官网)
(达咩)conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
正经事
3080的服务器依旧安装cuda11。在anaconda创建环境时,直接创建对应环境的python,然后conda install pytorch==1.X.X
, 就可以。以下两个是我实验的。可以用,做个记录。
Python3.8-》 Pytorch1.4.0 -》 Cuda101
Python3.7 -》 Pytorch1.3.1 -》 Cuda100
小师弟说, 官网代码pip装也好使,这个我没试 ,好用!!!
(不是专业的,以上仅个人经验,勿喷)
– 2022.5.2日更新
经过一年和3080的磨合。现在使用起来没什么问题。
3080有一个弊端,就是只能使用cuda11,
对于新手来说常常需要复现低版本cuda的代码,不一定要按照论文的一样环境配置,目的只是复现不是吗?
建议使用cuda11和对应版本的pytorch,98%是可以兼容的,个别函数会warning,但不影响代码运行。
这篇关于RTX3080ti Pytorch踩坑指南的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!