anaconda命令安装tensorflow-GPU,cuda,cudnn

1.首先创建tensorflow的虚拟环境,推荐python3.6.

在这里插入图片描述

2.安装cuda(conda命令为cudatoolkit),cudnn

然后根据需要查找tensorflow-GPU,cuda,cudnn的对应关系,附上链接https://developer.nvidia.com/zh-cn/cuda-gpus。
由于使用conda或者pip命令对于cuda,cudnn直接安装存在网络问题或者镜像源问题,要重复多次才能安装成功。
所以我选择直接下载好cudatoolkit,cudnn的安装包进行安装。安装包链接:https://repo.anaconda.com/pkgs/main/win-64/。安装包下载好了,cd切换到安装包的路径。
这里我下载的是

conda install cudatoolkit-10.0.130-0.tar.bz2
conda install cudnn-7.6.0-cuda10.0_0.tar.bz2

在这里插入图片描述
这种方式安装cuda,cudnn的方式十分舒畅便利!!!

3.tensorflow-GPU安装

上述的cuda,cudnn安装完成后,就是重头戏tensorflow-GPU的安装了。这里有两个安装方法:

3.1 anaconda navigator图形化安装
进入创建的虚拟环境,搜索tensorflow-gpu。
在这里插入图片描述在这里插入图片描述
选择自己需要的版本进行安装,这里提醒大家,如果一个版本安装之后不能使用,建议装相近的版本。例如我就试了1.15.0,1.14.0,1.13.0安装后使用import tensorflow as tf却不能使用,很有可能跟cuda,cudnn版本兼容的问题。最后试了1.10.0成功装上了,所以不要死磕一个版本,相近的版本也是可以使用的!!!安装完毕,进行测试GPU。

import tensorflow as tf
tf.test.is_gpu_available()

在这里插入图片描述
成功,可以愉快地跑实验了。

3.2 安装包安装
建议试了第一种方法不可以选择安装包安装,在这里安装包地址如下:http://mirrors.aliyun.com/pypi/simple/tensorflow-gpu/。
例:tensorflow_gpu-2.0.0a0-cp37-cp37m-win_amd64.whl
gpu-2.0版本,cp37代表python3.7版本,win_代表适用于windows系统
选择与本地环境相适应的版本下载。
在这里插入图片描述
参考:
笔记本GTX1650配置tensorflow-gpu2.2.0(利用anaconda命令安装cuda和cudnn
win10安装tensorflow-gpu版安装连接(visual studio2015+cudn8.0+cudnn6+anaconda3+tensorflow-gpu)
TensorFlow-gpu安装和测试(TensorFlow-gpu1.14+Cuda10)

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
对于安装 TensorFlow-GPU 1.14.0,您需要确保您的 Anaconda 环境已正确设置,并且已安装了正确版本的 CUDAcuDNN。 下面是安装步骤: 1. 首先,确保您已经成功安装Anaconda。如果还没有安装,请按照官方文档的指导进行安装。 2. 接下来,您需要安装 CUDA 10.0。您可以从 NVIDIA 官方网站下载适用于您的操作系统的 CUDA 安装程序,并按照指示进行安装。 3. 安装 CUDA 后,您需要安装 cuDNN 7.4。您需要先注册一个 NVIDIA 开发者账号,然后从 NVIDIA 开发者网站下载 cuDNN 安装包。根据您的操作系统和 CUDA 版本选择正确的版本,并根据 cuDNN 安装指南进行安装。 4. 安装CUDAcuDNN 后,打开 Anaconda Prompt(Windows)或终端(macOS/Linux)。 5. 创建一个新的环境并激活它: ``` conda create -n tf_gpu_env python=3.7 conda activate tf_gpu_env ``` 6. 在新环境中安装 TensorFlow-GPU 1.14.0: ``` pip install tensorflow-gpu==1.14.0 ``` 7. 安装完成后,您可以通过运行以下代码来验证 TensorFlow 是否正确安装使用 GPU: ```python import tensorflow as tf print(tf.test.is_gpu_available()) ``` 如果输出为 True,则表示 TensorFlow 正确安装并且可以使用 GPU。 请注意,安装 TensorFlow-GPU 需要确保您的系统满足 CUDAcuDNN 的要求,并且您的 GPU 支持 CUDA。如果您遇到任何问题,请参考 TensorFlow 官方文档或在相关论坛上寻求帮助。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值