我所做的是按照官方github网站上的说明
!git clone https://github.com/NVIDIA/apex
!cd apex
!pip install -v --no-cache-dir ./
它会给我一个错误:
ERROR: Directory './' is not installable. Neither 'setup.py' nor 'pyproject.toml' found.
Exception information:
Traceback (most recent call last):
File "/usr/local/lib/python3.6/dist-packages/pip/_internal/cli/base_command.py", line 178, in main
status = self.run(options, args)
File "/usr/local/lib/python3.6/dist-packages/pip/_internal/commands/install.py", line 326, in run
self.name, wheel_cache
File "/usr/local/lib/python3.6/dist-packages/pip/_internal/cli/base_command.py", line 268, in populate_requirement_set
wheel_cache=wheel_cache
File "/usr/local/lib/python3.6/dist-packages/pip/_internal/req/constructors.py", line 248, in install_req_from_line
"nor 'pyproject.toml' found." % name
pip._internal.exceptions.InstallationError: Directory './' is not installable. Neither 'setup.py' nor 'pyproject.toml' found.
8条答案
按热度按时间qrjkbowd1#
在添加CUDA_HOME环境变量后对我有效:
第一个
8cdiaqws2#
(本想加一条评论,但我没有足够的声望...)
它对我很有效,但实际上不需要
cd
。另外,我需要这里建议的两个全局选项:https://github.com/NVIDIA/apex/issues/86然后
r1zk6ea13#
已更新
首先,创建一个文件,例如
setup.sh
,如下所示:对于具有CUDA和C++扩展的apex:
然后安装
对于仅Python构建
仅Python版本省略了使用
apex.optimizers.FusedAdam
、apex.normalization.FusedLayerNorm
等所需的某些融合内核。检查顶点快速启动。
zkure5ic4#
在colab中,在
cd
命令前使用“%”而不是“!”上面的代码可以正常工作。
dfuffjeb5#
我尝试了几个选项,但我喜欢this website中的一个,它与fast_bert和torch一起工作得非常好:
oymdgrw76#
!cd apex
有问题。请改用%cd apex
。请阅读:https://stackoverflow.com/a/57212513/8690463
k7fdbhmy7#
我使用paperspace,这对我很有效:
xriantvc8#
2022年11月,以下内容对我有效。
apex.optimizers.FusedAdam
、apex.normalization.FusedLayerNorm
等需要CUDA和C++扩展(例如,请参见here)。因此,仅安装Python构建的版本是不够的。要构建apex
,PyTorch
和apex
的cuda版本必须匹配,如下所述。查询运行Ubuntu Colab的版本:
要获取当前cuda版本,请运行:
查找最新构建的PyTorch并计算平台here. x1c 0d1x
接下来,转到cuda toolkit archive并配置一个与PyTorch的cuda-version和您的OS版本匹配的版本。
复制安装说明:
删除
Sudo
并更改最后一行以包含您的cuda-version,例如!apt-get -y install cuda-11-7
(如果直接在shell中运行,则不带感叹号):您的cuda版本现在将被更新:
接下来,更新了Google Colab中过时的Pytorch版本:
构建apex。根据具体情况,您可能需要更少的全局选项:
现在可以根据需要导入顶点: