剛拿到實驗室一個6塊GPU的服務器賬號,無奈啥環境也沒有,需要自己配置一下,對於linux小白來說,走一步百度一步,紀錄一下流程。
因爲服務器是剛剛配置的,所以切換到root用戶時,顯示Authentication failure,需要手動改密碼進行切換,使用sudo passwd即可。su用來切換到root用戶,su - user01 用來切換到普通用戶user01
1.通過wget下載anconda安裝包:
wget https://repo.continuum.io/archive/Anaconda3-2018.12-Linux-x86_64.sh
2.利用bash命令安裝到anconda文件夾下:.
bash Anaconda3-2018.12-Linux-x86_64.sh -p anconda/ -u
一直按yes,包括添加到環境
如果運行conda --version 顯示沒有此指令的話,需要把anconda添加到變量中去,運行兩步:
echo 'export PATH="/home/user01/anaconda/bin:$PATH"' >> ~/.bashrc
source ~/.bashrc
3.添加清華的鏡像源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
並且在搜索時顯示搜索地址
conda config --set show_channel_urls yes
4.創建一個新的python環境:
conda create --name learning
5.查看當前CUDA和cudnn的版本號
cat /usr/local/cuda/version.txt
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
6.安裝VScode:
在官網下載Ubuntu對應的tar.gz,進行解壓和修改權限
tar xvzf code-stable-1552606946.tar.gz
有些教程中把vscode移動到了usr/local/目錄下,在進行權限修改
( mv VSCode-linux-x64 /usr/local/)
chmod +x /home/user01/vscode/VSCode-linux-x64/code
在終端運行code就能夠打開Vscode了
7.安裝pytorch:
在安裝pytorch之前,遇到了一個anaconda不是最新版本需要更新的問題,因爲anconda沒有權限,當直接使用
conda update -n base -c defaults conda 指令時會報507錯誤,說權限不夠,需要使用chomd指令來授權
sudo chmod -R 777 anaconda。完成授權之後,相應的按順序運行指令就可以了。
conda update -n base -c defaults conda
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
8.驗證pytorch是否安裝好了:
進入python,運行torch.cuda.is_available(),返回True的話就證明可以使用pytorch和GPU加速了