安装Tensorflow2.4.1 with Cuda 11.0 and CuDnn 8 Using Anaconda

1.安装CUDA

这个要看自己的显卡的适配
可以参考下面的文章
参考文章
https://blog.csdn.net/Hildasu/article/details/11408649
下面是CUDA下载的地方
CUDA
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_network
在这里插入图片描述

2.然后下载cuDNN

cuDNN Download

3.参考下面的文章

https://blog.csdn.net/Hildasu/article/details/114086497
完成将cuDNN里面的三个文件夹拖进去那个CUDA

4.找到自己的CUDA和python版本适合的Tensorflow

python版本在cmd中打上python就能看到
然后找到适合的Tensorflow
tensorflow各个版本需要的CUDA版本以及Cudnn的对应关系
相应的网址为:

https://www.tensorflow.org/install/source#common_installation_problems

https://www.tensorflow.org/install/source_windows

5.pip安装Tensorflow

现在是2021年4月22日,我的电脑是新的所以我直接用pip命令安装的
pip install tensorflow.
其他可参考
pip安装Tensorflow

cpu版本:pip install tensorflow==1.2.1 (==后面为所要安装的版本号)

gpu版本类似:pip install tensorflow-gpu==1.2.1

6.测试Tensorflow

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
print(tf.__version__)
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:', tf.test.is_gpu_available())

这里就证明GPU 调用成功了,cpu的肯定能跑
在这里插入图片描述

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值