RTX3060TI nvidia-tensorflow1.5安装

RTX3060TI nvidia-tensorflow1.5安装

新购买的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

https://developer.nvidia.com/blog/accelerating-tensorflow-on-a100-gpus/

posted @ 2020-12-27 12:34  船里的河  阅读(2489)  评论(0编辑  收藏  举报