最简单的安装tensorflow-gpu的方法(无需管理员权限,可以安装多个版本的tensorflow和cuda并且一一对应)

最简单的安装tensorflow-gpu的方法

利用conda安装(如果没有可以先安装Anaconda)

##conda create -n 你的虚拟环境名字 python=你想要的python版本##
conda create -n tensorflow python=3.7
##source activate 虚拟环境名字        ##这一步是进入你的虚拟环境##
source activate tensorflow
##以下尽量在你创建的虚拟环境中进行##
##conda install cudatoolkit=你tensorflow要安装的版本所对应的cuda版本
conda install cudatoolkit=10.0
conda install cudnn 
##conda install tensorflow-gpu=你想要的版本
conda install tensorflow-gpu=2.0

一定要注意cuda和cudnn的版本对应好你的tensorflow
liunx

我感觉这个方法很好用,不用动整个系统环境变量,只需要在虚拟环境中安装,所以可以不同的虚拟环境配置不同的tensorflow,pytorch等。而且不需要管理员权限。

如果你同时需要tensorflow1.14和1.13,就可以开两个虚拟环境,分别安装不同版本的cuda和cudnn,再安装对应的tensorflow13或14就好了

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值