1. 用conda創建虛擬環境,環境的名字叫base。
conda create -n base python=3.6
2. 進入剛剛創建的虛擬環境,base。
source activate base
如圖所示,已經進入了虛擬環境。
(base) gp@gp-System-Product-Name:~$
3. 查函cuda版本,方便安裝對應的深度學習框架Pytorch。
(base) gp@gp-System-Product-Name:~$ cat /usr/local/cuda/version.txt
CUDA Version 9.0.176
顯示我的cuda版本是9.0, 所以安裝cuda是9.0對應的版本。
4. 進入pytorch官網,查看安裝pytorch的命令。
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
開始一頓安裝,小夥伴請耐心等待,。。。
Proceed ([y]/n)? y
Downloading and Extracting Packages
python 3.6.10############################################################################### | 100%
mkl 2020.0################################################################################# | 100%
ninja 1.8.2################################################################################ | 100%
certifi 2020.4.5.1######################################################################### | 100%
libtiff 4.0.9############################################################################## | 100%
pytorch 1.1.0############################################################################## | 100%
libopenblas 0.3.9########################################################################## | 100%
numpy 1.18.1############################################################################### | 100%
cffi 1.14.0################################################################################ | 100%
libffi 3.2.1############################################################################### | 100%
setuptools 46.1.3########################################################################## | 100%
libblas 3.8.0############################################################################## | 100%
libcblas 3.8.0############################################################################# | 100%
llvm-openmp 10.0.0######################################################################### | 100%
torchvision 0.3.0########################################################################## | 100%
pillow 6.1.0############################################################################### | 100%
python_abi 3.7############################################################################# | 100%
sqlite 3.28.0############################################################################## | 100%
liblapack 3.8.0############################################################################ | 100%
ncurses 6.1################################################################################ | 100%
readline 7.0############################################################################### | 100%
_openmp_mutex 4.5########################################################################## | 100%
Preparing transaction: done
Verifying transaction: done
5. 安裝計算機視覺必要的包。
opencv 是處理圖像的包,tensorboardX可實現訓練可視化。
pip install opencv-python
pip install tensorboardX
pip install tensorflow
6. 檢查是否安裝成功。
python 進入python環境下。
Python 3.6.10 | packaged by conda-forge | (default, Apr 6 2020, 14:52:36)
[GCC 7.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import cv2
>>> import tensorboardX
>>> import tensorflow
如果能把對應的包import進來,說明你的環境配置成果了,開始訓練模型吧。
如果需要安裝其他的包,用命令: conda install 包名