前言
原本是使用的TensorFlow2.0,处理时间序列数据时发现一个很好用的函数:tf.keras.preprocessing.timeseries_dataset_from_array。不料报错没有此函数,才知道这个函数要TensorFlow2.3及以上才有,于是打算升级至TensorFlow2.3,过程中踩了几个坑,记录一下。
一、CUDA版本问题
原本使用的是CUDA10.0,看官网信息这样显示:TensorFlow 支持 CUDA® 10.1(TensorFlow 2.1.0 及更高版本)。想着一般CUDA是向下兼容的,于是直接把CUDA升级到了11.0,以免后面还要再升级CUDA。可是当我所有环境配置好之后显示这样的错误:
Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found
百思不得其解,查找资料发现CUDA版本只能是10.1,高一点都不行,遂重装。
这里分享一个小技巧,我们在下载CUDA时候网速会特别慢 ,我们只需要将下载链接复制到迅雷用迅雷下载就很快了。
二、GPU支持问题
CUDA版本问题解决了,开开心心运行TensorFlow,