Tnsorflow2.1安装教程,遇到的问题和解决办法
Tensorflow安装
首先需要预先安装Anaconda,使用Anaconda管理环境
安装好后在开始菜单中找到Anaconda Prompt
本文安装的为windows下GPU版本的tensorflow2.1
我们需要找好Tensorflow版本所对应的python,cuDNN及CUDA版本,否则安装完会因为版本问题T出现错误
对应版本可以在Tensorflow官网找到
创建新的环境
conda create -n 环境名 python=3.7
这里一定要加python=版本号,否则在pycharm选择环境时无法找打python.exe文件
创建成功后转到新创建的环境下
conda activate 环境名
删除环境方法
conda env remove -n/--name 环境名
安装CUDA
conda install cudatoolkit=10.1
安装CUDNN
conda install cudnn=6.7
安装Tensorflow
pip install tensorflow==2.1 -i https://pypi.douban.com/simple/
使用清华镜像下载会更快些
初步验证是否安装成功
pycharm设置
选择环境
新建一个项目
选择conda环境,可以看到这里的conda环境不是我们新创建的
找到Anaconda安装路径下的evns/新建环境名/python.exe文件
设置成功,进入项目
进一步验证是否安装成功
import tensorflow as tf
tensorfloa_vesion = tf.__version__
gpu_available = tf.config.list_physical_devices('GPU')
print("tensorflow version:", tensorfloa_vesion, "\GPU avalibale:", gpu_available)
a = tf.constant([1.0, 2.0], name="a")
b = tf.constant([1.0, 2.0], name="b")
result = tf.add(a, b, name='add')
print(result)
安装成功
安装遇到的问题
安装版本不匹配
由于安装的版本不匹配所以出现了以下错误
初步验证通过,pycharm运行程序出现
RuntimeError: cudaGetDevice()
都是由于 版本不匹配问题所引起的
驱动版本和运行驱动版本不匹配
其他都搞定,在进一步验证,返回错误
CUDA driver version is insufficient for CUDA runtime version
驱动版本无法满足运行版本
通过查看CPU驱动版本发现,本机的CPU版本为385…,而CUDA10.1版本要求最低为418.39,于是进行了驱动更新,问题解决。
具体对应版本如下所示,信息来源:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#title-resolved-issues
这篇好文章是转载于:学新通技术网
- 版权申明: 本站部分内容来自互联网,仅供学习及演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,请提供相关证据及您的身份证明,我们将在收到邮件后48小时内删除。
- 本站站名: 学新通技术网
- 本文地址: /boutique/detail/tanhfhicaf
系列文章
更多
同类精品
更多
-
photoshop保存的图片太大微信发不了怎么办
PHP中文网 06-15 -
Android 11 保存文件到外部存储,并分享文件
Luke 10-12 -
《学习通》视频自动暂停处理方法
HelloWorld317 07-05 -
word里面弄一个表格后上面的标题会跑到下面怎么办
PHP中文网 06-20 -
photoshop扩展功能面板显示灰色怎么办
PHP中文网 06-14 -
微信公众号没有声音提示怎么办
PHP中文网 03-31 -
excel下划线不显示怎么办
PHP中文网 06-23 -
excel打印预览压线压字怎么办
PHP中文网 06-22 -
怎样阻止微信小程序自动打开
PHP中文网 06-13 -
TikTok加速器哪个好免费的TK加速器推荐
TK小达人 10-01