码迷,mamicode.com
首页 > 其他好文 > 详细

RTX3060TI nvidia-tensorflow1.5安装

时间:2020-12-31 12:51:21      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:slow   model   ade   直接   install   app   uda   dal   blog   

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/

RTX3060TI nvidia-tensorflow1.5安装

标签:slow   model   ade   直接   install   app   uda   dal   blog   

原文地址:https://www.cnblogs.com/FreedomBox/p/14196893.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!