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.
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值