Windows系统无NVIDIA卡安装APEX

说明:Windows系统,无NVIDIA显卡无CUDA。

一、下载

下载地址:https://github.com/NVIDIA/apex
下载完成后解压后,得到源代码。

二、设置环境变量

如果直接安装会产生,因为没有NVIDIA卡驱动(CUDA),会产生如下错误:

Warning: Torch did not find available GPUs on this system.
 If your intention is to cross-compile, this is not an error.
By default, Apex will cross-compile for Pascal (compute capabilities 6.0, 6.1, 6.2),
Volta (compute capability 7.0), Turing (compute capability 7.5),
and, if the CUDA version is >= 11.0, Ampere (compute capability 8.0).
If you wish to cross-compile for a single specific architecture,
export TORCH_CUDA_ARCH_LIST="compute capability" before running setup.py.

如果要安装在Anaconda指定的环境(本文用py1指代),先用Anaconda Prompt激活py1环境。
在Anaconda Prompt上设置Windows系统环境变量,如下:

set TORCH_CUDA_ARCH_LIST="compute capability"

设置完成后,可以用set命令查询当前窗口所有的环境变量(如下),以确认是否设置成功。

set

注:set命令设置的环境变量只当前窗口有效。切记:后续操作在此Anaconda Prompt窗口进行。

三、安装APEX

在Anaconda Prompt窗口下先用cd命令切换至源代码根目录,然后执行如下命令即可完安装:

python.exe setup.py install

完成后用以下命令可以查看是否安装成功:

pip list|findstr apex

注意: python.exe setup.py install -v --no-cache-dir 反而会产生错误。

四、警告信息

如果没有安装CUDA或者CUDA版本小于11.0,则会出现如下警告信息,可忽略。

`fused_weight_gradient_mlp_cuda` module not found. gradient accumulation fusion with weight 
gradient computation disabled.
  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Transformer发轫于NLP(自然语言处理),并跨界应用到CV(计算机视觉)领域。 Swin Transformer是基于Transformer的计算机视觉骨干网,在图像分类、目标检测、实例分割、语义分割等多项下游CV应用中取得了SOTA的性能。该项工作也获得了ICCV 2021顶会最佳论文奖。 本课程将手把手地教大家使用labelImg标注和使用Swin Transformer训练自己的数据集。  本课程将介绍Transformer及在CV领域的应用、Swin Transformer的原理。 课程以多目标检测(足球和梅西同时检测)为例进行Swin Transformer实战演示。 课程在Windows和Ubuntu系统上分别做项目演示。包括:安装软件环境、安装Pytorch、安装Swin-Transformer-Object-Detection、标注自己的数据集、准备自己的数据集(自动划分训练集和验证集)、数据集格式转换(Python脚本完成)、修改配置文件、训练自己的数据集、测试训练出的网络模型、性能统计、日志分析。  相关课程: 《Transformer原理与代码精讲(PyTorch)》https://edu.csdn.net/course/detail/36697《Transformer原理与代码精讲(TensorFlow)》https://edu.csdn.net/course/detail/36699《ViT(Vision Transformer)原理与代码精讲》https://edu.csdn.net/course/detail/36719《DETR原理与代码精讲》https://edu.csdn.net/course/detail/36768《Swin Transformer实战目标检测:训练自己的数据集》https://edu.csdn.net/course/detail/36585《Swin Transformer实战实例分割:训练自己的数据集》https://edu.csdn.net/course/detail/36586《Swin Transformer原理与代码精讲》 https://download.csdn.net/course/detail/37045

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值