看板 DataScience 關於我們 聯絡資訊
各位大大好,小弟最近剛進碩班實驗室遇到一些問題想詢求幫助,若格式不妥會立即刪文。 實驗室因為研究需求配給我一台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://www.ptt.cc/bbs/DataScience/M.1629122109.A.F1C.html
liang1230: 自己build with CUDA 10試試看 08/16 23:04
yoyololicon: +1 這種只能自己照著readme build 08/17 00:36
followwar: 裝CUDA10自己bulid pytorch 08/17 20:47
wuyiulin: 先降 Ubuntu版本 16.04LTS 再用舊的CUDA 08/18 00:23
ajul1230: 感謝大大們熱心解答~ 08/18 16:26
yuwenche: 沒錯,這就是pytorch麻煩的地方.軟體不斷更新, 硬體也要 08/19 14:21
yuwenche: 跟著升級. 可用colab試試, 它在這方面蠻方便的. 08/19 14:25
yoyololicon: 跟ubuntu沒有關係 單純現在pre-build的pytorch不支援 08/19 16:43
ajul1230: 抱歉想再請問一下,所謂自己build PyTorch是指不靠conda 08/19 23:41
ajul1230: install之類的指令,而是直接去官網下載之類的嗎? 08/19 23:41
yoyololicon: 照這個吧 08/19 23:50
wuyiulin: 如果那台機器只有你自己用 pip管理 共用開conda建環境 08/19 23:54
ajul1230: 了解~再次感謝各位大大解惑 08/20 00:08