工作笔记｜百度AI studio配置pytorch环境

百度AI studio每日免费12小时v100的GPU显卡，可配置pytorch tensorflow环境！白漂它不香嘛！

• 新建一个项目，因为是随机分配服务器，所以每次使用pytorch tensorflow都需要先配置好环境
• 新建sh格式的shell文件，命名为(env.sh)，目的是导入清华源，加快python库的安装速度

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls yes

• 创建自己的虚拟环境

sh env.sh
conda create -n torch python=3.7
source activate torch
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.3.1 torchvision==0.2.2 pillow tqdm

• 新建py文件，写入torch代码，在终端中运行即可
• 终端nvidia-smi查看GPU使用情况！