1. 程式人生 > 其它 >Ubuntu18 系統下深度學習環境配置

Ubuntu18 系統下深度學習環境配置

技術標籤:ubuntu深度學習

目錄

1、顯示卡驅動+cuda11.1+cudnn8.0.4

【刪除cuda】

【安裝cuda】

【設定環境變數】

【CUDNN安裝】

2、Anaconda安裝

3、Anaconda配置

【建立虛擬環境】

【安裝pytorch】

【安裝opencv-python庫】

【安裝 pandas庫等】

【其他】

4、安裝 pycharm


1、顯示卡驅動+cuda11.1+cudnn8.0.4

此部分按照部落格:https://blog.csdn.net/weixin_43269994/article/details/109030404 安裝,驗證有效。

【刪除cuda

在命令列中輸入以下,刪除已安裝cuda:

sudo apt-get remove cuda
sudo apt --purge remove "*cublas*" "cuda*"
sudo apt-get autoclean
sudo apt-get remove cuda*

【安裝cuda

命令列輸入(紫色部分):

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin

#這一行下載cuda,對應1804版本,若報錯,也可直接在網站上輸入網址同樣可以下載。

sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600

#將下載的cuda檔案移動到新地址。

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda-repo-ubuntu1804-11-1- local_11.1.0-455.23.05-1_amd64.deb

#下載cuda-repo-ubuntu1804-11-1- local_11.1.0-455.23.05-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu1804-11-1-local_11.1.0-455.23.05-1_amd64.deb

#安裝

sudo apt-key add /var/cuda-repo-ubuntu1804-11-1-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

【設定環境變數】

新增路徑。

sudo vim ~/.bashrc
export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

使環境變數生效:

source ~/.bashrc

驗證結果:

nvcc -V

如果出現nvcc版本號,說明安裝成功。

CUDNN安裝】

進入:https://developer.nvidia.com/rdp/cudnn-download 下載“cuDNN Library for Linux”,下載完成後解壓tgz檔案,得到一個cuda資料夾:cudnn-11.1-linux-x64-v8.0.4.30.tgz

tar -xzvf cudnn-11.1-linux-x64-v8.0.4.30.tgz

然後複製檔案:(注意:部落格原文錯誤部分,在此處更正)

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

安裝結果驗證:

cat /usr/local/cuda-11.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

注意:我們之前所有配置的cuda環境都是在資料夾cuda-11.1中,結果如下:

2、Anaconda安裝

Linux下安裝anacondapytorch參考部落格:https://blog.csdn.net/hktxt/article/details/82111203

注意:選擇anaconda的版本和python版本對應。我的版本檢視到用的python3版本是3.6.9,對應anaconda版本為anaconda3-5.2.0。在國內的映象站點下載到對應版本:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

Anaconda3-5.2.0-Linux-x86_64.sh)

執行以下命令進行安裝,然後一路到底:

bash Anaconda3-5.2.0-Linux-x86_64.sh

安裝完anaconda後會建議安裝vscode,我在這裡遇到提示網路連結問題,可能是源的問題,這裡不重要,可採取其他方式安裝或者不安裝。

則安裝完成,輸入以下更新系統變數:

sudo gedit ~/.bashr

可見環境變數末尾添加了:

若沒有則手動新增。

3、Anaconda配置

【建立虛擬環境】

為pytorch建立一個虛擬環境,這是推薦的做法,這樣在建立其他環境如tensorflow時,就不會互相沖突。在終端輸入:

conda create -n pytorch python=3.6

這就建立了一個名為pytorch,python版本為3.6的虛擬環境。於是我們可以把Pytorch安裝在這個環境下面。首先我們啟用這個環境:source activate pytorch

source activate pytorch

啟用後,會看到前面會顯示(pytorch):

【安裝pytorch

在https://pytorch.org/ 官網,根據版本選擇,生成終端安裝命令。

一路安裝就好。

注意是在(名為pytorch)的環境下安裝。中途有遇到兩次http問題,導致跑到一半丟失的情況,因此重新安裝可以改善。

安裝完成。

在(pytorch)環境下(若沒進入環境需要按照之前說明啟用環境),進入python:

python

執行:

import torch 

沒報錯說明成功,退出環境記得關閉python:

exit()

退出環境:

source deactivate

【安裝opencv-python庫】

這裡用到的方法是在環境下運:

pip install opencv-contrib-python

【安裝 pandas庫等】

將工程程式碼放入到pycharm中後,發現報錯沒裝pandas庫,搜一下就知道了,注意安裝在執行環境下。

【其他】

*Anaconda操作說明:https://blog.csdn.net/lushuangning/article/details/110817844

可以conda list檢視已安裝的庫

4、安裝 pycharm

Pycharm是一個ide,推薦安裝專業版,而非免費的社群版,因為專業版可以ssh連遠端等附加功能。

安裝pycharm後,使用時記得設定執行環境,此處不贅述。