相信很多學深度學習的同學,都爲環境苦惱過,不想別人亂動自己辛苦配好的環境,想要一個單獨屬於自己的存儲空間,下面是我用過用過最簡單、安全、整潔的學習環境搭建方法,希望可以幫到你。
#Step1.home下用戶創建配置(這一步是在root環境下)
首先在home文件夾下創建用戶、設置密碼
#1創建用戶
sudo adduser yourname
sudo passwd yourname
#2給yourname添加root權限
打開visudo
添加 :
yourname ALL =(ALL:ALL) ALL
#Step2.cuda cudnn環境配置(此後幾步都是在你的用戶下)
這一步的前提是root環境下已經安裝好cuda cudnn
打開cuda安裝目錄下.bashrc文件,將cuda的環境變量拷貝到你用戶下的.bashrc文件中:
export PATH="/home/xxx/cuda9.0/bin:$PATH"
export LD_LIBRARY_PATH="/home/xxx/cuda9.0/lib64:$LD_LIBRARY_PATH"
#Step3.anconda 安裝
在你的用戶底下運行以下代碼:
bash "/home/xxx/software/Anaconda3-5.1.0-Linux-x86_64.sh"
#Step4.pytorch 安裝
或者直接運行以下安裝代碼:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
測試代碼:import torch
#Step5.tensorflow 安裝
安裝tensorflow用下面的代碼:
pip install tensorflow-gpu
安裝好之後用下列代碼測試一下:
import tensorflow as tf
tf.test.is_gpu_available()