在ubuntu20.04下搭建深度學習環境(pytorch1.5)


  首先聲明,完成這個環境的搭建不是我一個人的努力,是我在網上查找好多相關資料並得益於他們的幫助完成的,在下文對應的地方會放上相關鏈接。整個過程我將它分為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的安裝,在此對以上引用到博文的博主一並表示感謝!

  最后,歡迎大家提出疑問,批評指正!

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM