Pytorch2.1在昇腾服务器的环境配置与使用
收藏回复举报
Pytorch2.1在昇腾服务器的环境配置与使用
发表于2023-11-02 10:16:47
0 查看

  今年10月份,pytorch官网的技术文档上写明已经与昇腾有了深入合作,在全新版本的Pytorch2.1中,将全面支持昇腾系列服务器,这件事表明,在后续pytorch的生态将与华为的昇腾系列服务器产生更紧密的关系,而用户在使用过程中也将更加方便,之前在英伟达的项目也将能够更加平滑的迁移到昇腾的生态中。

本次,主要讲明如何在昇腾服务器中配置全新的torch2.1,并且能够调用npu中的算力。首先,我们需要安装全新的驱动与固件,在此处,我们必须使用的固件版本为7.0.RC1 alpha005版本,这是16号发布的全新版本。

cke_8385.png

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

cke_10183.png

以上是一些相关的工具依赖,最好先把这些依赖安装完毕,然后再进行驱动的安装,安装完驱动之后,再进行固件的安装。

然后安装最新版本的CANN,版本号为7.0.RC1。

cke_11741.png

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

cke_13300.png

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

20220109.png

本帖最后由 匿名用户2023/11/02 11:40:30 编辑

我要发帖子