您好,想问下现在只有基于python3.6的rocm版本pytorch吗,有没有基于python3.8及以上的?
目前还没有,解决方法如下:1、在自己账号目录下安装python3.8及以上版本,并export python3路径;2、 https://download.pytorch.org/whl/torch_stable.html 下载所需的torch whl; 3、加载相应的rocm版本,确认python3路径,安装torch whl文件。
感谢回复,我按照您的提示conda创建一个py3.9的虚拟环境,在~/.bashrc中export了python路径,然后下载torch和torchvision,在此虚拟环境中安装,但后面报错得不到cuda,请问是怎么回事?
请使用rocm4.0.1或dtk21.04吧,驱动是rocm4.0的,跟rocm4.1不兼容,后续我们把rocm4.1优化一下
想问下您使用python3.9是有什么要求吗?我们平台目前支持比较好的是rocm4.0.1 python3.6环境的pytorch。module load apps/PyTorch/1.9.0a0/pytorch_1.9-rocm_4.0.1即可
目前还没有,解决方法如下:1、在自己账号目录下安装python3.8及以上版本,并export python3路径;2、 https://download.pytorch.org/whl/torch_stable.html 下载所需的torch whl; 3、加载相应的rocm版本,确认python3路径,安装torch whl文件。