今年10月份,pytorch官网的技术文档上写明已经与昇腾有了深入合作,在全新版本的Pytorch2.1中,将全面支持昇腾系列服务器,这件事表明,在后续pytorch的生态将与华为的昇腾系列服务器产生更紧密的关系,而用户在使用过程中也将更加方便,之前在英伟达的项目也将能够更加平滑的迁移到昇腾的生态中。
本次,主要讲明如何在昇腾服务器中配置全新的torch2.1,并且能够调用npu中的算力。首先,我们需要安装全新的驱动与固件,在此处,我们必须使用的固件版本为7.0.RC1 alpha005版本,这是16号发布的全新版本。

在安装驱动之前,因为它有一些gcc的依赖,所以我们最好先安装服务器环境的一些依赖项,防止安装失败。

以上是一些相关的工具依赖,最好先把这些依赖安装完毕,然后再进行驱动的安装,安装完驱动之后,再进行固件的安装。
然后安装最新版本的CANN,版本号为7.0.RC1。

安装完毕之后,再安装python环境,此处推荐的python版本为3.8及以上。安装完python之后,再安装python的相关依赖。

此时前期的环境就已经配置完毕,我们只需要通过pip安装pytorch2.1,同时值得开心的是全新版本的torch_npu也可以通过pip直接安装,只需要输入命令pip3 install torch-npu==2.1.0rc1。此时,所有的环境均已配好。接下来进行验证。

今年10月份,pytorch官网的技术文档上写明已经与昇腾有了深入合作,在全新版本的Pytorch2.1中,将全面支持昇腾系列服务器,这件事表明,在后续pytorch的生态将与华为的昇腾系列服务器产生更紧密的关系,而用户在使用过程中也将更加方便,之前在英伟达的项目也将能够更加平滑的迁移到昇腾的生态中。
本次,主要讲明如何在昇腾服务器中配置全新的torch2.1,并且能够调用npu中的算力。首先,我们需要安装全新的驱动与固件,在此处,我们必须使用的固件版本为7.0.RC1 alpha005版本,这是16号发布的全新版本。
在安装驱动之前,因为它有一些gcc的依赖,所以我们最好先安装服务器环境的一些依赖项,防止安装失败。
以上是一些相关的工具依赖,最好先把这些依赖安装完毕,然后再进行驱动的安装,安装完驱动之后,再进行固件的安装。
然后安装最新版本的CANN,版本号为7.0.RC1。
安装完毕之后,再安装python环境,此处推荐的python版本为3.8及以上。安装完python之后,再安装python的相关依赖。
此时前期的环境就已经配置完毕,我们只需要通过pip安装pytorch2.1,同时值得开心的是全新版本的torch_npu也可以通过pip直接安装,只需要输入命令pip3 install torch-npu==2.1.0rc1。此时,所有的环境均已配好。接下来进行验证。