新购买的RTX3060TI在尝试多次后发现只能使用CUDA11,对应的也只能用tensorflow2.4。参考相关帖子实现CUDA11+tensorflow1.5。
依赖包：
直接pip install 安装会报错，下载下来再安装。
nvidia-cublas 11.2.1.74
nvidia-cuda-cupti 11.1.69
nvidia-cuda-nvrtc 11.1.74
nvidia-cuda-nvcc 11.1.74
nvidia-cuda-runtime 11.1.74
nvidia-cudnn 8.0.4.30
nvidia-cufft 10.3.0.74
nvidia-curand 10.2.2.74
nvidia-cusolver 11.0.0.74
nvidia-cusparse 11.2.0.275
nvidia-dali-cuda110 0.26.0
nvidia-dali-nvtf-plugin 0.26.0+nv20.10
nvidia-nccl 2.7.8
nvidia-pyindex 1.0.5
nvidia-tensorboard 1.15.0+nv20.10
nvidia-tensorrt 7.2.1.4
tensorflow-estimator 1.15.1
python==3.6
keras==2.3.1
h5py==2.10
注意：
跑代码报错时，使用pip list | grep 查看原有的tensorflow是否未卸载，将非nvidia-tensorflow的tenforslow包卸载。
参考链接：
https://blog.csdn.net/wu496963386/article/details/109583045?utm_source=app