创建python环境:
conda create -n tf25 python=3.8
进入python环境:
conda activate tf25
安装tensorflow2.5
pip install tensotflow-gpu==2.5
使用ConvLSTM3D这个类,需要在tf2.5基础上安装以下模块
pip install tf-nightly-gpu==2.7.0.dev20210630
查看CUDA版本
将nvcc目录添加到PATH(和tf无关)
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin
cuDNN安装
下载后解压并将对应文件复制到cuda文件夹里
转载请注明来源 https://tianweiye.github.io