深度學習--pytorch安裝


一、查看cuda及cudnn版本

  先確保安裝了顯卡:nvidia-smi 查看

  

  cat /usr/local/cuda/version.txt

 

  cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2  或者 nvcc --version ,后一個版本更靠譜,因為前者完全依賴/usr/local/cuda/include/cudnn.h 這個文件,而這個文件並不一定非得在這個路徑中,尤其是既沒有root又沒有admin權限的情況下。

  下載 cudnn包,一般 cuda 8.0版搭配的都是 cudnn 6.0版本。

  我下載的是:cudnn-8.0-linux-x64-v6.0.tgz,一般官網可以下載,網盤也有別人分享的資源,官網下載更快些,只是需要注冊用戶,不過注冊也很方便啦。

  解壓后把lib64的路徑添加至LD_LIBRARY_PATH即可,對了,確保cuda的路徑(一般默認:/usr/local/cuda/bin)已在PATH中。然后執行nvcc --version基本就可以查看到版本信息了。

  • PATH環境變量下面是找可執行命令的位置 
     
  • LD_LIBRARY_PATH這個是gcc編譯的時候找需要鏈接的庫的位置  

  

 

 

 

二、官網鏈接

  http://pytorch.org/

  支持三種安裝方式,conda,pip 以及source

   

  我選擇的是源碼安裝,鏈接為:https://github.com/pytorch/pytorch 但是安裝的時候發現cmake沒裝,重新安裝cmake。

  資源下載:https://cmake.org/download/

  教程:https://geeksww.com/tutorials/operating_systems/linux/installation/downloading_compiling_and_installing_cmake_on_linux.php

  這篇教程說的很詳細,但是奈何沒有admin權限,更遑論root了,只能安裝到自己目錄下,后來發現最后一步可以設置安裝路徑:

    
$ make install DESTDIR="/some/absolute/path"
 或
$ export DESTDIR="/some/absolute/path"
$ make install

 

   安裝完成

  注意此時還需要將cmake的安裝路徑添加到PATH中去,export PATH=/home/XXX/usr/local/bin:$PATH

     然后敲cmake即可發現已安裝完畢,或者直接cmake -version即可查看安裝版本。

 

 

  然后直接Python setup.py install即可

  

  一個小提示:安裝完成后,在當前的pytorch文件夾中輸出Python,import會報錯:ModuleNotFoundError: No module named 'torch._C'

  主要原因是該文件夾與包重名,退出這個文件夾后重新import即可。

  參考鏈接:https://github.com/pytorch/pytorch/issues/574

 

  附需要了解的python包,很好的進度條:tqdm

  參考鏈接:http://spaces.ac.cn/archives/3902/

 


免責聲明!

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



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