
PyTorch 1.4 with ROCM 3.1 (AMD GPU) on Ubuntu 18.04
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目提供在Ubuntu 18.04系统上安装PyTorch 1.4和ROCm 3.1的详细教程,支持AMD GPU加速深度学习计算。
对于AMD GPU用户来说,现在可以用AMD GPU来学习人工智能了。pytorch 1.4与ROCM 3.1在ubuntu 18.04系统上已经预编译好,可以直接通过pip install xxxx.whl命令安装使用。只需提前安装好rocm平台驱动即可轻松开始使用。
全部评论 (0)
还没有任何评论哟~


