深度学习 环境配置(Tensorflow,Pytorch)多版本CUDA共存

本文介绍了如何在Anaconda环境下管理深度学习库Tensorflow和Pytorch的安装,以及如何在多版本CUDA之间切换。在CUDA和cuDNN已安装的前提下,创建特定Python版本的环境,然后分别安装Tensorflow和Pytorch的不同版本。此外,还讨论了如何在激活的环境中使用超级用户权限来管理这些库。

建议在anaconda下创建环境,再进行安装,方便各个环境的管理,以下安装方法在cuda和cudnn都安装好的前提下进行的。

Tensorflow安装

  1. 首先创建环境:conda create -n tf_0.11_gpu python=2.7; tf_0.11_gpu为环境的名称,2.7指定了python的版本,根据需要可以换成3.6; 环境名称可以随意修改
  2. 激活当前环境: source activate tf_0.11_gpu
  3. 在该环境中安装tensorflow:
    pip install --ignore-installed --upgrade
    https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/ 这个网站有tensorflow 各个版本,根据需要进行解决

Pytorch安装

  1. 首先按照上面步骤创建环境并进行激活;
  2. 根据Pytorch官网要求进行安装,具体网址如下:
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值