[問題] 舊版GPU執行PyTorch問題

看板DataScience作者 (保鏢)時間2年前 (2021/08/16 21:55), 編輯推噓2(2012)
留言14則, 6人參與, 2年前最新討論串1/1
各位大大好,小弟最近剛進碩班實驗室遇到一些問題想詢求幫助,若格式不妥會立即刪文。 實驗室因為研究需求配給我一台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

08/16 23:04, 2年前 , 1F
自己build with CUDA 10試試看
08/16 23:04, 1F

08/17 00:36, 2年前 , 2F
+1 這種只能自己照著readme build
08/17 00:36, 2F

08/17 20:47, 2年前 , 3F
裝CUDA10自己bulid pytorch
08/17 20:47, 3F

08/18 00:23, 2年前 , 4F
先降 Ubuntu版本 16.04LTS 再用舊的CUDA
08/18 00:23, 4F

08/18 16:26, 2年前 , 5F
感謝大大們熱心解答~
08/18 16:26, 5F

08/19 14:21, 2年前 , 6F
沒錯,這就是pytorch麻煩的地方.軟體不斷更新, 硬體也要
08/19 14:21, 6F

08/19 14:25, 2年前 , 7F
跟著升級. 可用colab試試, 它在這方面蠻方便的.
08/19 14:25, 7F

08/19 16:43, 2年前 , 8F
跟ubuntu沒有關係 單純現在pre-build的pytorch不支援
08/19 16:43, 8F

08/19 23:41, 2年前 , 9F
抱歉想再請問一下,所謂自己build PyTorch是指不靠conda
08/19 23:41, 9F

08/19 23:41, 2年前 , 10F
install之類的指令,而是直接去官網下載之類的嗎?
08/19 23:41, 10F

08/19 23:50, 2年前 , 11F

08/19 23:50, 2年前 , 12F
照這個吧
08/19 23:50, 12F

08/19 23:54, 2年前 , 13F
如果那台機器只有你自己用 pip管理 共用開conda建環境
08/19 23:54, 13F

08/20 00:08, 2年前 , 14F
了解~再次感謝各位大大解惑
08/20 00:08, 14F
文章代碼(AID): #1X6cuzyS (DataScience)
文章代碼(AID): #1X6cuzyS (DataScience)