发布时间:2024-06-16 13:01
打开cmd输入:pip install tensorflow-gpu
(默认是安装最新版,如果需要指定安装版本,例如2.5.0版本,则可以输入:pip install tensorflow-gpu==2.5.0
)
理论上来说下载11.1或以上版本的CUDA都可以,这里以11.7为例:
点击进入官网下载
如图所示:
Win10用户可以点击下面链接直接下载:
点击下载cuda11.7
点击官网下载cudnn
需要登录NVIDIA账号之后才能下载,嫌麻烦的,点击下面链接可以直接下载:
点击直接下载cudnn8.3.3
下载完成之后解压好,把文件名改成cudnn。然后放进刚刚安装好的CUDA的文件夹里,如下图所示:
接下来就是把cuda和cudnn添加进环境变量。一共是放两个:
(注意:添加完后最好把他们都移到所有环境变量的最上边,防止出BUG)
按照以上教程配置好之后,如果运行程序,可能会报这个错:
Could not locate zlibwapi.dll. Please make sure it is in your library path!
Process finished with exit code -1073740791 (0xC0000409)
这是因为缺少了zlibwapi导致的,点击下载zlibwapi
下载好并解压后,放到C:\Windows\System32
文件夹内,然后添加下面路径到环境变量:C:\Windows\System32\zlib123dllx64\dll_x64
,最后重启电脑!!!
按照上述的步骤配置好后,GPU就可以正常使用了!
在TensorRT中使用动态形状(Dynamic Shapes)
亚马逊测评成长之路!跌跌撞撞有了具体的成就!现在给大家分享一下!
【历史上的今天】7 月 15 日:Mozilla 基金会正式成立;Enigma 密码机的首次工作;任天堂推出 FC 游戏机
stm32单片机+amg8833+红外热成像/单片机红外测温成像/stm32 amg8833红外热成像
做⼀个⾼德地图的 iOS / Android MAUI 控件(下)
Docker的卸载清除;报错rm:Device or resource busy