cuda drive與cuda api 版本不一樣導致找不到 libcublasLt.so.12

cuda pytorch

alanotmt 2023-04-13 09:44:02 ‧ 933 瀏覽

分享至

小弟是ubuntu 22.04跑
以下是硬軟環境:
Nvidia GPU 4080， CUDA 11.7 cuDNN 8.8.1
由於4080目前要用到cuda12的drive 所以我是裝12的drive
但pytorch只支援到11.8的cuda版本，我是裝11.7啦

後來去裝一些pytorch插件就發現找不到libcublasLt.so.12
那是肯定的阿，我是裝cuda11 只有libcublasLt.so.11沒有12的
這樣的情況下，我要怎麼做才能讓他辨識到我cuda api其實是11不是12?

ccutmis iT邦高手 2 級 ‧ 2023-04-13 10:03:14 檢舉

有參考過這個嗎?
https://pytorch.org/get-started/previous-versions/

alanotmt iT邦新手 4 級 ‧ 2023-04-13 13:46:17 檢舉

安裝過程都沒有有出現error或warning 但就跑辨識時就會彈could not load library libcublaslt.so.12.

登入發表討論

直播研討會

{{ item.channelVendor }} {{ item.webinarstarted }} |

直播中

1 個回答

增廣建文

iT邦研究生 5 級 ‧ 2023-04-16 00:21:54

Pytorch有支援CUDA 11.6 & 11.7的wheel

# CUDA 11.6
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
# CUDA 11.7
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia

真的還是不行話可以自己重build pytorch

回應
分享
檢舉

登入發表回應

我要發表回答

立即登入回答

參賽組數

1064 組

團體組數

40 組

累計文章數

22182 篇

完賽人數

597 人

15th鐵人賽 16th鐵人賽 13th鐵人賽 14th鐵人賽 12th鐵人賽 11th鐵人賽鐵人賽 2019鐵人賽 javascript 2018鐵人賽 python 2017鐵人賽 windows php c# windows server linux css react vue.js

IT邦幫忙