本文共 415 字,大约阅读时间需要 1 分钟。
使用amd 的gpu进行深度学习训练,安装完rocm和tensorflow 1.x 后,会提示
ImportError: libmcwamp.so.2: cannot open shared object file: No such file or directory
这是因为rocm 升级到3.1后,动态库名称变了,需要作个符号链接作兼容处理
ln -s /opt/rocm-3.1.0/lib/libmcwamp.so.3 /opt/rocm-3.1.0/lib/libmcwamp.so.2
环境:
ubuntu 18.04
rocm 3.1
tensorflow 1.15
硬件:
intel i7 6700
AMD Radoen VII (平民法 拉利 雷七,可战2080ti, 16G大显存,妈妈再也不用担心我的显存 不够用了)
结论: rocm 3.1 + 预编译的tf 1.15.2 工作正常,其它版本测试不过。
转载地址:http://aswl.baihongyu.com/