ubuntu16.04 配置 pytorch 和 tensorflow 独立环境

Ubuntu 16.04 配置pytorch 和 tensorflow 独立环境

  1. 安装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” 使操作生效。

  2. 安装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”

  3. 安装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”

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值