首先聲明,完成這個環境的搭建不是我一個人的努力,是我在網上查找好多相關資料並得益於他們的幫助完成的,在下文對應的地方會放上相關鏈接。整個過程我將它分為4個步驟。
一、在ubuntu20.04上安裝python3.7;
不知道為什么,我下載的這個ubuntu20.04沒有自帶python環境,剛開始我也驚了一下,不過不管那么多了,裝一下也還好啦。故自行安裝python3.7,簡單粗暴:sudo apt-get install python。它會自動安裝最新版python。
二、檢查硬件並安裝對應的顯卡驅動;
參考鏈接如下:https://blog.csdn.net/qiancaobaicheng/article/details/95096354
首先使用ubuntu-drivers devices命令查看自己的機器所需要安裝的驅動版本,比如我這個TITAN RTX(型號)就需要安裝430或者440。然后參照上面鏈接中的信息,在官網上下載對應的驅動並安裝即可。
注:目前最新的ubuntu20.04系統在選擇驅動的時候,選用ubuntu18.04或者19.04的都可以。
三、安裝cuda、cudnn並檢查是否正確安裝;
關於cuda、cudnn的安裝,這篇博文寫得很好,大家可以參考安裝https://blog.csdn.net/ashome123/article/details/105822040/。我在這里就不再贅述了。但需要注意的是中間的gcc降級,一般可能會想不到這個,因為在windows下應該是不需要這個操作的(據我所知)。
成果圖:
四、利用清華鏡像安裝pytorch1.5。
為什么要用清華鏡像安裝pytorch1.5呢?因為官網直接在線安裝提示timeout(連不上);又沒有whl文件供下載(晚點應該會有)。所以只能考慮用別的鏡像源了,最終在網上找到了這個。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=10.0
原諒鏈接:https://blog.csdn.net/zzq060143/article/details/88042075
因為我安裝的是cuda10.2,所以我將上述最后一行的參數“cudatoolkit=10.0”修改為“cudatoolkit=10.2”就可以了,也是支持GPU的。不知道評論區里說的不支持GPU是什么情況,感謝這位博主。
截圖為證:
終於站在巨人的肩膀上,完成了ubuntu20.04環境下深度學習環境python3.7.7、cuda、cudnn、pytorch1.5的安裝,在此對以上引用到博文的博主一並表示感謝!
最后,歡迎大家提出疑問,批評指正!