windows安装tensorflow GPU

2024-05-16 11:42:10

一、安装Anaconda

Anaconda是Python发行包，包含了很多Python科学计算库。它是比直接安装Python更好的选择。

二、安装Tensorflow

如果安装了tensorflow，默认是CPU版的，要想安装GPU版的，必须先pip uninstall tensorflow卸载干净之后，再使用pip install tensorflow-gpu来安装。

tensorflow CPU版比较简单，安装之后直接运行就可以。
tensorflow GPU版需要依赖CUDA，具体安装见下文。

三、CUDA和cudnn的安装

GPU是并行计算中必不可少的东西，在这一领域Nvdia处于领先地位。神经网络这种计算量巨大、适合并行化，正是需要GPU的场景。
Tensorflow GPU版依赖CUDA，而cudnn是连接tensorflow和CUDA的纽带。所以需要去Nvdia官网上下载CUDA和cudnn。CUDA安装包比较大，大概1G多；cudnn安装包下载前需要注册用户填写调查问卷。需要注意的是，CUDA和cudnn的版本必须得匹配。
另外，Tensorflow用到的cudnn库可能不是最新的，所以如果下载最新版的cudnn很有可能tensorflow无法识别。

下载完cudnn之后，解压得到三个文件夹bin、include、lib，将其覆盖复制粘贴到CUDA的安装目录下即可。
要想让tensorflow看见cudnn.dll，就要让CUDA的bin目录添加到PATH环境变量中去。

CUDA安装完成后，需要重启电脑。

可能遇到的异常

Cannot remove entries from nonexistent file c:\anaconda\lib\site-packages\easy-install.pth

解决方案

pip install --upgrade --ignore-installed setuptools

Blas SGEMM launch failed

解决方案：关闭其他Python交互窗口

转载于:https://www.cnblogs.com/weiyinfu/p/6973670.html

windows安装tensorflow GPU相关推荐

windows安装TensorFlow和Keras遇到的问题及其解决方法
windows安装TensorFlow和Keras遇到的问题及其解决方法参考文章: (1)windows安装TensorFlow和Keras遇到的问题及其解决方法 (2)https://www.cn ...
通过Anaconda在Ubuntu16.04上安装 TensorFlow(GPU版本)
一. 安装环境 Ubuntu16.04.3 LST GPU: GeForce GTX1070 Python: 3.5 CUDA Toolkit 8.0 GA1 (Sept 2016) cuDNN v6 ...
解决Windows安装TensorFlow报错：ERROR: Cannot uninstall 'wrapt'问题
错误描述报错信息如下图所示, 报错原因: 在安装的过程中,检测到了wrapt包已经存在,但是又因为不能确定哪些文件属于这个包而无法把它卸载.于是就出现了报错,导致TensorFlow安装不成功. 问 ...
linux安装tensorflow教程,Ubuntu 16.04 安装 TensorFlow(GPU支持)
本文记录Ubuntu 16.04安装Tensorflow步骤,也包括怎么从源码编译安装Tensorflow. 要想安装Tensorflow GPU版本,你需要有一个新一点的Nvidia显卡. Tens ...
windows10下安装tensorflow(gpu版本)
windows10安装tensorflow的gpu版本(pip3安装方式) 前言: TensorFlow 有cpu和 gpu两个版本:gpu版本需要英伟达CUDA 和 cuDNN 的支持,cpu版本不 ...
Docker 安装 TensorFlow GPU 实战
2019独角兽企业重金招聘Python工程师标准>>> Docker 安装 TensorFlow GPU 实战博客分类: 深度学习 docker 安装背景 AI如雨后春笋般的出现, ...
GTX860M安装Tensorflow gpu踩坑
GTX860M安装Tensorflow gpu踩坑前几天给台式成功配置了lightgbm的gpu版本和tensorflow的gpu版本,但是在单位还是要用LSTM,写一篇对应版本的踩坑文章日后查阅. ...
Ubuntu 20.04 安装Tensorflow GPU版本（GTX-1060显卡）
Ubuntu 20.04 安装Tensorflow GPU版本 (GTX-1060显卡) 参考: https://cyfeng.science/2020/05/02/ubuntu-install-nv ...
Windows 下无痛安装 TensorFlow (GPU版)
Windows 下无痛安装 TensorFlow 2 (GPU版) 一. 原材料硬件 CUDA算力>=3.5的NVIDIA显卡如果想看你的显卡是否符合要求, 移步至 https://deve ...

最新文章

热门文章