logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

tensorflow-gpu版本环境安装及测试

在保证tensorflow基本环境没问题下,进行tf-gpu版本安装。step1- 版本要确定英伟达与cuda版本对应:来源:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html要确定tensorflow与cuda版本对应:来源:https://tensorflow.google.cn/install/source_w

#tensorflow#深度学习
问题:cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

在尝试使用gpu加速计算时遇到这个问题,在这里做个记录。在安装tensorflow_gpu后,使用代码:print("GPU is", "available" if tf.config.experimental.list_physical_devices("GPU") else "NOT AVAILABLE")检测gpu是否能够参与计算。代码输出为:GPU is available但是在执行先前的

#深度学习#tensorflow
到底了