RTX3080ti Pytorch踩坑指南

RTX3080ti Pytorch踩坑指南

**

2022.5.2日更新在最后,建议先看

**

  1. 写在前面的罗里吧嗦

都说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

RTX3080ti Pytorch踩坑指南_第1张图片

正经事

3080的服务器依旧安装cuda11。在anaconda创建环境时,直接创建对应环境的python,然后conda install pytorch==1.X.X, 就可以。以下两个是我实验的。可以用,做个记录。

Python3.8-》 Pytorch1.4.0 -》 Cuda101

Python3.7 -》 Pytorch1.3.1 -》 Cuda100
RTX3080ti Pytorch踩坑指南_第2张图片
小师弟说, 官网代码pip装也好使,这个我没试 ,好用!!!

(不是专业的,以上仅个人经验,勿喷)

– 2022.5.2日更新

经过一年和3080的磨合。现在使用起来没什么问题。
3080有一个弊端,就是只能使用cuda11,
对于新手来说常常需要复现低版本cuda的代码,不一定要按照论文的一样环境配置,目的只是复现不是吗?
建议使用cuda11和对应版本的pytorch,98%是可以兼容的,个别函数会warning,但不影响代码运行。

你可能感兴趣的:(pytorch,深度学习,python)