準備測試的代碼是:
https://github.com/jwyang/faster-rcnn.pytorch
講解:https://hellozhaozheng.github.io/z_post/PyTorch-FasterRCNN/
一、查看Ubuntu版本
- 使用命令:
lsb_release -a
輸出:
jingge@svr1:~$ lsb_release -a
No LSB modules are available.
Distributor ID: Ubuntu
Description: Ubuntu 16.04.5 LTS # 我的版本16.04
Release: 16.04
Codename: xenial
二、查看linux系統位數,以下載對應anaconda版本
- 使用命令
getconf LONG_BIT
輸出64 - 或者使用命令
uname -a
輸出
Linux svr1 4.4.0-91-generic #114-Ubuntu SMP Tue Aug 8 11:56:56 UTC 2017 x86_64 x86_64 x86_64 GNU/Linux
# x86_64說明我是64位內核, 跑的是64位的系統;
# i386, i686說明你是32位的內核, 跑的是32位的系統
三、anaconda下載安裝
-
想用python3.6版本,參考https://blog.csdn.net/yuejisuo1948/article/details/81043823
準備下載Anaconda3-5.2.0 -
關閉VPN,進入清華鏡像源:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
-
選擇Anaconda3-5.2.0-Linux-x86_64.sh,下載到自己的H盤下新建的Anaconda3-5.2.0文件夾中
-
在自己下載的目錄下,運行
bash Anaconda3-5.2.0-Linux-x86_64.sh
- 一直enter直到出現
Please answer 'yes' or 'no'
,輸入yes
- 出現默認安裝目錄,確定按照它的默認
/anaconda3
,所以enter
- 安裝中installing
installation finished
後,出現Do you with the installer to prepend the Anaconda3 install location to PATH in your /home/username/.bashrc?
,輸入yes- 出現
Thank you for installing Anaconda3!
- anaconda安裝完成
Do you wish to proceed with the installation of Microsoft VSCode
?詢問我是否裝vscode,我選擇yes安裝,迅速完成提示installed
- 安裝完成後 不要忘記輸入
source ~/.bashrc
(否則仍舊爲Ubuntu默認的python2.7)
- 再輸入
python -V
,輸出顯示python 3.6.5:: Anaconda, Inc
參考
https://blog.csdn.net/zaishijizhidian/article/details/81663387
https://blog.csdn.net/weixin_40294256/article/details/79555606
四、創建一個和已經裝了的python3.6共存的python2.7的anaconda環境
- 輸入
conda create -n py27 python=2.7
- 以後在使用中
- 使用source activate py27激活python2.7環境
- 使用source deactivate關閉python2.7環境
五、查看CUDA和cudnn版本
- 查看cuda 版本
- CUDA 8.0.61
cat /usr/local/cuda/version.txt
- 查看 CUDNN 版本
- 5.1.10
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
六、安裝配置pytorch
- 參考
https://blog.csdn.net/leviopku/article/details/84548816
- 使用conda安裝新的虛擬環境,用於這次pytorch0.4.0,python3.6的配置。
- 不必每次都對tf,pytorch等升降版本。
- 還可以對虛擬環境安裝額外的包。
更多指令參考:
https://blog.csdn.net/lyy14011305/article/details/59500819
conda create -n py36pytorch python=3.6
- 驗證安裝好了一個虛擬環境:
$ source activate py36pytorch
(py36pytorch) jingge@svr1:~$ python -V
Python 3.6.8 :: Anaconda, Inc.
- 進入pytorch官網https://pytorch.org/get-started/locally/ 選擇自己CUDA8.0,Linux,conda,stable,python3.6的配置,出現一個命令
conda install pytorch torchvision cudatoolkit=8.0 -c pytorch
- 使用上述命令安裝pytorch 0.4.0
- 發現結果安裝的不是自己想要的pytorch0.4.0:
- 卸載上面版本的pytorch:
conda uninstall pytorch
conda uninstall libtorch
- 考慮到上次的方法直接安裝特別慢,還是找了清華源的pytorch的鏡像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/?C=N&O=A
- 選擇好cuda和pytorch版本在瀏覽器下載
- 用conda install 解壓安裝,驗證版本,完成:
今天get新技能:
① linux中用ls打開一個目錄後,如何快速進入裏面
使用cd 進入目錄,輸入想進入的目錄的前幾個字母,然後按tab鍵即可自動補全。
②從別處複製後,在putty軟件中鼠標右鍵即可粘貼
明天終於可開始下載數據跑代碼了,
記住數據放在hdd2T中!
2019.4.11修改
昨天發現作者的是基於python2.7編譯的,爲了保持一致除去不必要的麻煩,我之前的虛擬環境py36pytorch
配置的是python3.6,需要重新配置一個python2.7的。在這個過程中,經歷了以下麻煩:
- 新建
py27pytorch
虛擬環境,在清華鏡像源下載安裝的pytorch,conda install
完成後import torch
總是報錯,pytorch安裝不成功 - 重新再之前建立了沒用的py27環境中安裝,一樣的錯誤
- 重裝anaconda:
rm -rf anaconda3
bash Anaconda3-5.2.0-Linux-x86_64.sh
- 沒有用
source ~/.bashrc
命令,直接關閉終端重新開一個。 conda -V
驗證anaconda安裝成功
- 新建名爲pytorch的虛擬環境,配置python2.7
conda create -n pytorch python=2.7
- 激活環境
conda activate pytorch
- 在該環境下,去pytorch官網下載(而不是像之前一樣去清華鏡像源),雖然速度慢:
網址: https://pytorch.org/get-started/locally/
找到自己需要版本的命令:
conda install pytorch torchvision cudatoolkit=8.0 -c pytorch
-下載完成,驗證
import torch
torch.cuda.is_available()
torch.__version__ # 重開終端後,似乎需要用source activate而不是conda activate 激活環境
- 刪除之前的不要的虛擬環境
conda env remove -n py27
conda env remove -n py27pytorch
conda env list