apex库安装,解决amp not installed 报错( ModuleNotFoundError: No module named ‘apex‘)

问题描述:
在这里插入图片描述
这是缺少apex库,需要安装!注意:不要使用pip install apex,虽然也能成功安装,但是最后运行程序会发现依旧出错,用不了

具体安装步骤如下:

# 执行git下载apex文件夹时,如果下载太慢,可以手动进入网址https://github.com/NVIDIA/apex下下载解压后执行后续操作
git clone https://github.com/NVIDIA/apex
cd apex
python setup.py install

执行后如果出现下图所示结果,表示安装成功!!
在这里插入图片描述

  • 补充
# error
Traceback (most recent call last):
File "setup.py", line 35, in <module>
_, bare_metal_major, _ = get_cuda_bare_metal_version(CUDA_HOME)
File "setup.py", line 14, in get_cuda_bare_metal_version
raw_output = subprocess.check_output([cuda_dir + "/bin/nvcc", "-V"], universal_newlines=True)
TypeError: unsupported operand type(s) for +: 'NoneType' and 'str'

是因为conda安装的cuda是不全的,没有nvcc,解决方案conda install -c nvidia cuda-nvcc,安装成功!!

记得收藏哟,防止下次找不到啦!
看在我这么认真的份上,点个赞再走吧

  • 24
    点赞
  • 43
    收藏
    觉得还不错? 一键收藏
  • 11
    评论
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值