概述
自己研究，制作不宜，转发请标注来源
使用环境：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
第六步：完成
最后
以上就是单纯画笔为你收集整理的使用AMD显卡安装tch-rs(Rust版Pytorch)的全部内容，希望文章能够帮你解决使用AMD显卡安装tch-rs(Rust版Pytorch)所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复