使用AMD显卡安装tch-rs(Rust版Pytorch)

自己研究，制作不宜，转发请标注来源

使用环境：6700xt,opensuse15.4,rocm5.4

接着上一篇的，如果有不懂的先看上一篇

第一步：安装libtorch(直接下pytorch官网的，解压)

第二步：安装一个pytorch rocm5.1的(如果你的rocm版本小于rocm5.2则不用管)

第三步：根据tch-rs github上的页面配置
 

export LIBTORCH=/path/to/libtorch
export LD_LIBRARY_PATH=${LIBTORCH}/lib:$LD_LIBRARY_PATH
export LIBTORCH_INCLUDE=/path/to/libtorch/
export LIBTORCH_LIB=/path/to/libtorch/

第四步：在cargo.toml添加依赖

[dependencies]

tch="0.9.0"

第五步：软链接文件夹

(运行它会告诉你哪里缺少个TensileLibrary.dat，直接切到那个路径)

(缺文件夹就直接mkdir)

ln -s ~/anaconda3/envs/<刚刚安的5.1版本conda那个包名>/lib/<那个包的python版本>/site-packages/torch/lib/library library

第六步：完成