ubuntu 1604安裝cuda 10.1 ,cudnn 10.1,pytorch 1.3.1 ,libtorch

文章目錄

• • • • 1、安裝cuda10.1
      • 2、安裝cudnn 10.1
      • 3、安裝pytorch
      • • 3.1 libtorch
        • 3.2 pytorch

1、安裝cuda10.1

從這裡下載下傳cuda10.1的toolkits

cuda 10.1下載下傳連結

按下圖選擇

安裝步驟就按照Base Installer中的四句指令來安裝，第二句的指令cuda-repo-<version> 需要修改，在執行彎第一句指令後terminal log的最後一句裡，直接複制粘貼到指令行。cuda toolkit會幫忙我們将nvidia驅動也一并安裝好。安裝完之後在終端視窗中輸入：nvidia-smi指令，如果顯示如下，則說明cuda已經安裝好了。

可能會有用的連結黑屏問題解決 https://blog.csdn.net/qq_39670011/article/details/90404111 ，Ubuntu16.04+cuda10.1官方安裝指導的補充說明。這篇文章中說需要對顯示卡驅動做一些設定，否則重新開機電腦可能會黑屏，不過我們的電腦并沒有出現所說的問題，暫且将網址記錄到這裡，如果出現問題可以參考解決。

2、安裝cudnn 10.1

下載下傳cudnn 10.1，官網：cudnn下載下傳位址 https://developer.nvidia.com/rdp/cudnn-archive。選擇對應cuda 10.1的版本。我選擇了最上面的這個

安裝方法參考：cudnn安裝方法 https://blog.csdn.net/suezhang9/article/details/98657431。這個連結裡說了兩種安裝方法，我按第一種安裝方法安裝了一下。就是将對應的頭檔案和庫檔案拷貝到了指定的目錄下。

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
           

3、安裝pytorch

3.1 libtorch

打開官網 https://pytorch.org/，因為我要用C++開發，是以按下圖選擇，按照”Download here (cxx11 ABI)“的連結下載下傳然後解壓到自己指定的目錄下。

這是libtorch官方安裝使用說明：安裝連結 https://pytorch.org/cppdocs/installing.html

使用的時候需要在自己的CMakeLists.txt中添加下面兩句cmake指令，然後包含頭檔案和lib檔案

list(APPEND CMAKE_PREFIX_PATH /path/to/libtorch/share/cmake/Torch)
find_package(Torch REQUIRED)
include_directories(${TORCH_INCLUDE_DIRS})


add_executable(${PROJECT_NAME} your cpp files)

targe_link_libraries(${PROJECT_NAME} ${TORCH_LIBRARIES})
           

3.2 pytorch

安裝python版的pytorch，按下圖選擇。我的python版本是3.6。這裡簡單的按照自己的python版本選擇就可以。

使用pip3安裝的時候遇到了問題，是pip3的問題，修改一下/usr/bin目錄下的pip3腳本如下，并儲存。然後就可以使用pip3進行安裝了。

然後執行：

sudo pip3 install torch torchvision
           

安裝就結束了。