Windows安装tensorflow-gpu=1.14.0CUDA=10.0cuDNN=7.4 (多版本CUDA共存)
最佳答案 问答题库578位专家为你答疑解惑
文章目录
- 0. 前置说明
- 1. 查看版本对应关系
- 2. 安装 cuda
- 3. 安装 cudnn
- 4. 添加环境变量
- 5. 安装 tensorflow
0. 前置说明
本机(Windows 11
)已安装CUDA 11.7
使用命令查看显卡驱动:
nvidia-smi
这里显示的CUDA Version: 11.7
说明支持安装11.7
版本及以下的CUDA
,如果需要安装高版本的需要先更新显卡驱动
本文介绍安装tensorflow-gpu 1.14.0
的环境配置,需要先安装CUDA 10.0
版本以及cuDNN 7.4
版本
1. 查看版本对应关系
tensorflow 版本对应关系可以查看 https://tensorflow.google.cn/install/source_windows?hl=zh-cn
本文下载tensorflow-gpu 1.14.0
版本,需要安装cuda 10.0
版本以及cudnn 7.4
版本
2. 安装 cuda
cuda
下载地址: https://developer.nvidia.com/cuda-toolkit-archive
第一步有精简安装
和自定义安装
两个选项,记得选自定义安装
接下来在选项中仅勾选CUDA
,并去掉CUDA
选项下Visual Studio Integration
的勾选
(这一步的选项非常重要,如果勾选了其他的可能导致高版本已安装的CUDA
被覆盖)
这里使用默认的安装位置就可以,可以记一下安装位置,后面需要用到
3. 安装 cudnn
cudnn
下载地址: https://developer.nvidia.com/rdp/cudnn-archive
发现对于Windows
系统并没有直接显示cudnn
的版本,可以把鼠标放在上面,在左下角会显示版本号,这里选择安装适用于cuda 10.0
的7.4.2.24
版本,应该只要是7.4
版本都可以的
下载之后解压,将文件夹下的桑文件夹复制到cuda10.0
安装的根目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
4. 添加环境变量
可以看到系统环境变量已经自动添加了cuda 10.0
:
将以下四个路径添加到Path
环境变量中:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
其中默认已经添加了bin
和lib
的路径,我猜测这一步不是必须要做的hhh
打开终端,验证是否安装成功,输入命令:
nvcc -V
可以看到已经成功切换到10.0
版本的CUDA
了
(注意如果原本已经打开了终端,需要关闭重新打开)
5. 安装 tensorflow
创建新的虚拟环境,指定python
安装版本:
conda create -n tf python==3.7
tensorflow-gpu 1.14.0
版本可以支持3.5-3.7
版本的python
,这里安装3.7
版本
安装tensorflow-gpu
:
pip install tensorflow-gpu==1.14.0
经过漫长的等待,终于安装成功:
验证是否正确安装并可以使用GPU:
python
import tensorflow as tf
print(tf.test.is_gpu_available())
对于第一个import
语句,会输出很多行warning
,第二个语句我第一次运行一直没有输出最后的True
还是False
,还以为安装出错了,结果第二次运行成功输出True
:
如果在Pycharm
终端显示的cuda
版本还是原来的,需要重启Pycharm
配环境实在不易,给自己鼓个掌,加油~
(接下来还有更难的在等你😔)
99%的人还看了
相似问题
- flink1.13.6版本的应用程序(maven版)
- 技术分享| gcc版本升级到5.2
- 基于 Glibc 版本升级的 DolphinDB 数据查询性能优化实践
- 解决requests 2.28.x版本SSL错误:证书验证失败
- UE5和UE4版本更新重大改变汇总。
- Docker build报错总结,版本过新大避雷!
- K8S1.23.5部署(此前1.17版本步骤囊括)及问题记录
- Mongodb3.4升级高版本mongoTemplate.executeCommand报错The cursor option is required
- onnx模型转换opset版本和固定动态输入尺寸
- centos安装指定版本docker
猜你感兴趣
版权申明
本文"Windows安装tensorflow-gpu=1.14.0CUDA=10.0cuDNN=7.4 (多版本CUDA共存)":http://eshow365.cn/6-29799-0.html 内容来自互联网,请自行判断内容的正确性。如有侵权请联系我们,立即删除!
- 上一篇: springboot源码阅读需要提前了解的知识点
- 下一篇: 机器学习(六)构建机器学习模型