作者ajul1230 (保镖)
看板DataScience
标题[问题] 旧版GPU执行PyTorch问题
时间Mon Aug 16 21:55:07 2021
各位大大好,小弟最近刚进硕班实验室遇到一些问题想询求帮助,若格式不妥会立即删文。
实验室因为研究需求配给我一台Ubuntu 20.04系统主机搭配GTX Titan Black显卡,我也按照指示安装了CUDA 11.1以及相对应的cuDNN。
最近实验需要使用到PyTorch,但我执行後始终会显示以下bug:
CUDA error: no kernel image is available for execution on the device
我上网爬了一些文似乎是我的显卡运算力只有3.5,而新版的PyTorch似乎不再支援此运算力的显卡。我改安装PyTorch1.2後确实能执行,但因为比较旧版仍有些不方便的地方。
我实验室的学长也没有遇过这种状况帮不太了我。因此想在这里询问看看有没有了解的大大我的显卡是否能运行新版本的PyTorch...由於刚接触这块领域很多都不太熟,再麻烦指教了,十分感谢。
-----
Sent from JPTT on my Asus ASUS_I002D.
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 42.73.20.146 (台湾)
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/DataScience/M.1629122109.A.F1C.html
1F:→ liang1230: 自己build with CUDA 10试试看 08/16 23:04
2F:→ yoyololicon: +1 这种只能自己照着readme build 08/17 00:36
3F:→ followwar: 装CUDA10自己bulid pytorch 08/17 20:47
4F:推 wuyiulin: 先降 Ubuntu版本 16.04LTS 再用旧的CUDA 08/18 00:23
5F:→ ajul1230: 感谢大大们热心解答~ 08/18 16:26
6F:→ yuwenche: 没错,这就是pytorch麻烦的地方.软体不断更新, 硬体也要 08/19 14:21
7F:→ yuwenche: 跟着升级. 可用colab试试, 它在这方面蛮方便的. 08/19 14:25
8F:→ yoyololicon: 跟ubuntu没有关系 单纯现在pre-build的pytorch不支援 08/19 16:43
9F:→ ajul1230: 抱歉想再请问一下,所谓自己build PyTorch是指不靠conda 08/19 23:41
10F:→ ajul1230: install之类的指令,而是直接去官网下载之类的吗? 08/19 23:41
12F:→ yoyololicon: 照这个吧 08/19 23:50
13F:推 wuyiulin: 如果那台机器只有你自己用 pip管理 共用开conda建环境 08/19 23:54
14F:→ ajul1230: 了解~再次感谢各位大大解惑 08/20 00:08