Linux自动配置深度学习环境

628 阅读1分钟

最近总是要配置深度学习环境(python==3.6+tf gpu==1.13.1+ keras + pytorch==1.1.0),直接写了一个脚本如下:

#下载和安装anaconda
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.2.0-Linux-x86_64.sh
bash Anaconda3-5.2.0-Linux-x86_64.sh

#切换到清华源
wget https://tuna.moe/oh-my-tuna/oh-my-tuna.py
python oh-my-tuna.py

#安装深度学习框架
conda install tensorflow-gpu==1.13.1
conda install keras
conda install pytorch torchvision cudatoolkit=10.0

#安装系统查看器
pip install glances

附上一些conda常用的操作:

conda环境创建、退出和移除

conda create --name py36 python=3.6 #创建环境
conda activate py36 #激活环境
conda deactivate py36 #退出环境
conda remove --name py36 --all #删除环境
conda info -e #查看系统中的所有环境

conda查找包

conda search tensorflow

conda恢复默认源

conda config --remove-key channels

conda瘦身

conda clean -p      //删除没有用的包
conda clean -t      //清理tar