在虚拟环境里通过anaconda安装tensorflow-gpu

下载对应版本的anaconda

launch CMD.exe.prompt

输入命令建立一个虚拟环境:conda create -n tensorflow python=x.x

输入命令进入虚拟环境:conda activate tensorflow

输入命令下载ipython:pip install ipython 【注意:要在虚拟环境中下载】

输入命令下载jupyter:pip install jupyter notebook

输入命令下载tensorflow-gpu【tensorflow2.0 之后gpu和cpu应该是一个包】:pip install tensorflow-gpu

输入pip list能看到安装的tensorflow

去anaconda.org上搜索cuda和cudnn对应版本的安装包Cudnn :: Anaconda.org,然后在cmd

里运行conda install -c esri cudnn和conda install -c esri cuda,如果下载不了就直接下载安装包

然后把cudnn下载解压后结果复制到cuda解压的文件里,再把cuda文件复制到D:\anaco\envs\tensorflow里anaco是anaoconda的所在文件,如下图是复制后的tensorflow文件夹。

注意python3.7-10对应cuda 11.2和cudnn8.1.0,其他对应情况可以去tensorflow官网查询在 Windows 环境中从源代码构建  |  TensorFlow

国际官网数据更全,上面是中文网站

在tensorflow虚拟环境中输入jupyter notebook 启动jupyter notebook

新建folder后打开文件

输入import tensorflow as tf

tf.test.is_gpu_available()

运行

返回true即说明返回可以调用gpu且tensorflow安装成功

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值