天天看點

Pytroch入坑 1. Deep15.5 + CUDA8.0 + pytroch環境搭建

0.前言:

一個Pytroch簡單入門系列,配置 i5-6200u+ gt940m+4G記憶體+deepin15.5.

Ubuntu應用雖廣,但是配置環境好麻煩....以後用熟了再換吧....小白上手即用推薦Deepin,什麼都弄好了。

推薦安裝教程網址:https://blog.csdn.net/puzzleofmyheart/article/details/79183094(桌上型電腦)

https://blog.csdn.net/liaodong2010/article/details/71482304

1.安裝nvidia驅動

現在絕大多數筆記本都是雙顯示卡,本文推薦的方法适合絕大多數筆記本使用者,桌上型電腦參照上面的網址。

1.安裝nvidia-bumblebee

sudo apt update

sudo apt install nvidia-smi nvidia-driver

一行指令搞定nvidia驅動、bumblebee切換程式、和顯示卡狀态監控程式。

不用管nouveau驅動,系統會自己屏蔽掉。

然後重新開機

sudo reboot

檢測一下

Pytroch入坑 1. Deep15.5 + CUDA8.0 + pytroch環境搭建

需要注意,安裝完nvidia-bumblebee後,以後針對顯示卡的操作前面都要加上  optirun

2.  安裝cuda驅動

deepin友善之處來了。

如果式cuda8.0及其以下,deepin自帶的gcc版本高了,需要進行降級。

sudo apt install g++-4.8 gcc-4.8

cd /usr/bin

sudo rm gcc g++

sudo ln -s g++-4.8 g++

sudo ln -s gcc-4.8 gcc

這個操作過程中可能會報錯,無視,執行完就行

安裝cuda的開發工具

sudo apt install nvidia-cuda-dev nvidia-cuda-toolkit nvidia-nsight nvidia-visual-profiler

sudo apt install python-pycuda   

python-pycuda 是cuda的python開發接口。

3. 下載下傳pytorch

到pytorch的官網上去下載下傳pytorch 的包

Pytroch入坑 1. Deep15.5 + CUDA8.0 + pytroch環境搭建

一種是根據自己的需要,選擇安裝指令再終端中自動下載下傳安裝,但是國内環境較差,比如我這個連指令都刷不出來,是以選擇離線安裝的方式:

http://pytorch.org/previous-versions/

在此網頁選擇自己适合的安裝檔案

我下的是torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl,最好選擇帶mu的,我開始選帶m的不相容

下好後執行

sudo pip install -0.3.0.post4-cp27-cp27mu-linux_x86_64.whl

pytroch 若安裝失敗,則可能是版本不對、pip沒更新、numpy沒更新,百度一下就行

4.測試

Pytroch入坑 1. Deep15.5 + CUDA8.0 + pytroch環境搭建

5.torchvision安裝

這是官方的用來下載下傳模型,加載資料的,很友善

我用 pip安裝失敗,最後參照這篇部落格  https://blog.csdn.net/guotong1988/article/details/79065646

git clone https://github.com/pytorch/vision.git 

git checkout 0.2.0 

pip install pillow 

python setup.py install

可能會讓你更新pip,照做就行。安裝完這個就能跑大多數代碼了~