Tensorflow-Gpu安装(踩坑以及部分常见错误,dll缺少错误等)

发布于:2022-10-14 ⋅ 阅读:(474) ⋅ 点赞:(0)
  1. 首先第一步是查看自已电脑的Gpu显卡,先去英伟达官网获取适合自已的驱动
    1. 获取好后在cmd输入nvidia-smi就会输出cuda核心
      1.         
  2. 安装适合自已的CUDA toolkit 和 cuDNN 不安装这两个直接运行tensorflow会显示缺少dll文件然后就Cpu的跑、
    1. CUDA Toolkit Archive | NVIDIA Developericon-default.png?t=M85Bhttps://developer.nvidia.com/cuda-toolkit-archive 这个比较繁琐就自行研究了反正最后安上就行
    2. CUDA Toolkit 11.8 Downloads | NVIDIA DeveloperResources CUDA Documentation/Release NotesMacOS Tools Training Sample Code Forums Archive of Previous CUDA Releases FAQ Open Source PackagesSubmit a BugTarball and Zip Archive Deliverableshttps://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64
    3. https://developer.nvidia.com/rdp/cudnn-downloadicon-default.png?t=M85Bhttps://developer.nvidia.com/rdp/cudnn-downloadCuda Dnn
  3. 安装好后就是tensorflow-gpu版本的安装
    1. python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.6.0这个支持cuda11.x版本具体下哪个版本可以上tensorflow那参考对应关系
  4. 最后就是Dll缺少的问题
    1. 重安cudatool cudadnn还是没效果
    2. 我最后是从其他电脑拿cudatool 和 dudadnn 里边的所有dll文件复制到c盘的Windows/System32 路径下就好了
本文含有隐藏内容,请 开通VIP 后查看

网站公告


今日签到

点亮在社区的每一天
去签到