深度学习服务器自动配置小脚本

组里买了新的Server，分给多用户用，但是每次创建新用户配置环境都很麻烦，所以搞了个小脚本。需要自取哟~


#!/bin/bash
install_dir=/home/${USER}/anaconda3

DIR="/packages"  # script dir
bash $DIR/Anaconda3-2020.11-Linux-x86_64.sh -b -p $install_dir

# initialzation for conda virtual env
$install_dir/bin/conda init

echo "Anaconda 安装完成，下次登录即可使用conda虚拟环境"
echo "==============================================="
echo "安装 PyTorch, torchvision..."

$install_dir/bin/pip install /packages/torch-1.7.0-cp38-cp38-linux_x86_64.whl
$install_dir/bin/pip install /packages/torchvision-0.8.1-cp38-cp38-linux_x86_64.whl

# test installation
$install_dir/bin/python -c "import torch"
if [[ $? == "0" ]]; then
    echo "PyTorch 安装成功"
else
    echo "PyTorch 安装失败，请重试。"
    exit 1
fi

$install_dir/bin/python -c "import torchvision"
if [[ $? == "0" ]]; then
    echo "torchvision 安装成功!"
else
    echo "torchvision 安装失败，请重试!"
    exit 1
fi

flag=0

until ((flag==1))
do
    echo "输入一个用于连接code-server的端口(0~9999): "
    read port

    pIDa=`netstat -an | grep LISTEN | grep 0.0.0.0:$port\ `
    echo $pIDa
    if [ "$pIDa" == "" ];then
        flag=1
    else
        flag=0
        echo "端口已经被使用 请换一个："
    fi
done

echo "请设置code-server密码: "
read vspassword

echo "code-server配置文件已经存储到 ~/.config/code-server/config.yaml"

mkdir -p /home/${USER}/.config/code-server

touch /home/${USER}/.config/code-server/config.yaml
echo "bind-addr: 0.0.0.0:$port
auth: password
password: $vspassword
cert: false" > /home/${USER}/.config/code-server/config.yaml

cp /packages/ms-python-release.vsix /home/${USER}
cp /packages/VisualStudioExptTeam.vscodeintellicode-1.2.11.vsix /home/${USER}

mkdir -p /home/${USER}/.local/share/code-server/User/
touch /home/${USER}/.local/share/code-server/User/Settings.json
echo "{
    \"extensions.autoCheckUpdates\": false,
    \"extensions.autoUpdate\": false
}" > /home/${USER}/.local/share/code-server/User/Settings.json

echo "正在启动code-server ..."
nohup code-server > code-server.log 2>&1 &

echo "配置完成，可在 http://59.69.103.185:$port 查看code server"
echo "登陆密码：$vspassword"
echo "=================================="
echo "Python扩展：ms-python-release.vsix"
echo "IntelliCode扩展： VisualStudioExptTeam.vscodeintellicode-1.2.11.vsix"

脚本中的一些文件出处：

Anaconda：https://repo.anaconda.com/archive/Anaconda3-2020.11-Linux-x86_64.sh

Torch的安装包：https://download.pytorch.org/whl/cu102/torch-1.7.0-cp38-cp38-linux_x86_64.whl

Torchvision的安装包：https://download.pytorch.org/whl/cu102/torchvision-0.8.1-cp38-cp38-linux_x86_64.whl

PS: 采用1.7和cuda10.2的原因：写这个脚本的时候，1.8还没有出来，服务器上装的是10.2的cuda（没钱买30系）

Code-server的设置：直接去github下载deb包安装就可以：
https://github.com/cdr/code-server/releases

Python能用的扩展：https://github.com/microsoft/vscode-python/releases/tag/2020.10.332292344
Intellicode的扩展：https://marketplace.visualstudio.com/items?itemName=VisualStudioExptTeam.vscodeintellicode

fin