AI人工智能
pytorch版本问题
2021-12-03 11:16:05
201

您好,想问下现在只有基于python3.6的rocm版本pytorch吗,有没有基于python3.8及以上的?

全部评论(7)
只看楼主
T
tianlh
1楼

目前还没有,解决方法如下:1、在自己账号目录下安装python3.8及以上版本,并export python3路径;2、 https://download.pytorch.org/whl/torch_stable.html 下载所需的torch whl; 3、加载相应的rocm版本,确认python3路径,安装torch whl文件。

2021/12/3
A
ac490q7j1h
回复 @tianlh

感谢回复,我按照您的提示conda创建一个py3.9的虚拟环境,在~/.bashrc中export了python路径,然后下载torch和torchvision,在此虚拟环境中安装,但后面报错得不到cuda,请问是怎么回事?

2021/12/3
T
tianlh
回复 @tianlh

请使用rocm4.0.1或dtk21.04吧,驱动是rocm4.0的,跟rocm4.1不兼容,后续我们把rocm4.1优化一下

2021/12/6
T
tianlh
回复 @tianlh

想问下您使用python3.9是有什么要求吗?我们平台目前支持比较好的是rocm4.0.1 python3.6环境的pytorch。module load apps/PyTorch/1.9.0a0/pytorch_1.9-rocm_4.0.1即可

2021/12/6
全部6条回复
没有更多内容了
--
楼主
Level.4
0 帖子
楼主热帖
暂无数据