记录Tensorflow2.4.0版本的安装
硬件配置和软件版本
硬件配置: Win10+3060Ti
软件版本: Python3.6+Tensorflow1.14.0+CUDA10.0+cudnn7.6.5
但是安装上述的版本运行程序会报如下错误:
failed to run cuBLAS routine: CUBLAS_STATUS_EXECUTION_FAILED
在网上搜索了一下好像30系显卡只支持CUDA11及以上的CUDA版本,因此将软件版本更新为:
软件版本: Python3.6+Tensorflow2.4.0+CUDA11+cudnn8.0.5
软件安装
Tensorflow2.4.0安装
conda create -n tensorflow python=3.6
activate tensorflow
# 利用清华镜像源安装tensorflow,tensorflow2.x版本不区分cpu和GPU版本
pip install tensorflow