深度学习apex包安装失败的问题

文章目录

  • 一、前言
  • 二、正确的安装方法
    • 2.1 从github上下载
    • 2.2 切换到下载后的路径
    • 2.3 进行安装

一、前言

在进行论文复现的时候,发现缺了一个重要的包:apex包。

这是NVIDIA开发的基于PyTorch的混合精度训练加速神器,能够增加运算速度,并且减少显存的占用。

但是安装失败:

深度学习apex包安装失败的问题_第1张图片

二、正确的安装方法

2.1 从github上下载

git clone https://github.com/NVIDIA/apex.git

2.2 切换到下载后的路径

cd apex

2.3 进行安装

python setup.py install

深度学习apex包安装失败的问题_第2张图片

下载成功!

你可能感兴趣的:(错误或者警告类型解决办法总结,深度学习笔记,软件安装教程,深度学习,python,apex)