使用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 工作正常,其它版本测试不过。

Logo

免费领 200 小时云算力,进群参与显卡、AI PC 幸运抽奖

更多推荐