Ubuntu 16.04 配置pytorch 和 tensorflow 独立环境
-
安装Anaconda
1.1 下载anaconda镜像,推荐使用中科大源
https://mirrors.ustc.edu.cn/anaconda/archive/
本次选用"Anaconda3-5.3.1-Linux-x86_64.sh"
1.2 安装anaconda到指定位置
在上步下载的镜像文件夹下运行 “bash Anaconda3-5.3.1-Linux-x86_64.sh -p ./anaconda3/ -u”指令,这样即可将anaconda 安装在“ ./anaconda3/”目录下。
1.3 安装成功后,键入“source ~/.bashrc” 使操作生效。 -
安装pytorch独立环境
2.1 创建pytorch虚拟环境
打开终端,键入“ conda create -n pytorch_env python=3.7” 指令,后续根据提示,全程输入y,即可创建名为“pytorch_env”的虚拟环境,其中环境默认的python版本为3.7.
2.2 键入“conda env list”,即可查看已创建的虚拟环境列表。
2.3键入“source activate pytorch_env”,即可进入已创建的虚拟环境。
2.4 安装pytorch
键入“conda install pytorch torchvision cudatoolkit=8.0 -c pytorch”即可,我目前的机器是cuda8.0的,所以上指令对应8.0,大家可以根据自己机器的cuda版本修改。安装过程中根据提示,全程输入y即可。
安装完成后,检查是否安装成功,键入“python”,然后“import torch”,如果没有错误输出,即安装成功,继续键入“torch.version”即可查看安装的pytorch 版本。
2.5 安装其他相关第三方库,根据需要安装即可。
键入“pip install pandas numpy matplotlib jupyter notebook”
2.6 退出虚拟环境,只需键入“source deactivate pytorch_env” -
安装tensorflow独立环境
创建适合tensorflow运行的独立环境与pytorch独立环境类似
3.1 创建tensorflow虚拟环境
键入“conda create -n tf_env python=3.7”
3.2 安装tensorflow
键入“ pip install --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple”
3.3 安装相关第三方库
键入“pip install pandas numpy matplotlib jupyter notebook”