NVIDIA / NVIDIA/apex

ModuleNotFoundError: No module named 'fused_adam_cuda'

Open
#633 9 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
9k
Forks
1.5k
Avg merge
2d 4h
Merged PRs (30d)
3

Description

hello,ModuleNotFoundError: No module named 'fused_adam_cuda'

linux
pytorch 3.7
cuda9.2

Has been an error:ModuleNotFoundError: No module named 'fused_adam_cuda'。

git clone https://github.com/NVIDIA/apex
$ cd apex
$ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

$` pip install -v --no-cache-dir ./

python setup.py install --cuda_ext --cpp_ext

注释掉setup.py中版本检查那段,都试过了,还是有错
Comment out the check of version in setup.py, i've tried everything above, there are still mistakes。 No module named 'fused_adam_cuda'。

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reviewing setup.py and reproducing the reported installation commands on Linux with the stated PyTorch and CUDA versions. Check why the build does not provide fused_adam_cuda; the issue is done when the installation completes and importing that module succeeds.

Written by the indexing model from the issue text.

Assessment

Tech stack
linux, python, pytorch
Domain
build-system, machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.