NVIDIA / NVIDIA/apex

Installation errors

Open
#1,507 1 comment 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

I am using cuda11.1 pytorch1.8.0 ubuntu20.04 to install apex using the following command:
git clone https://github.com/NVIDIA/apex
cd apex
Export TORCH_CUDA_ARCH_LIST = "6.0; 6.1; 6.2; 7.0; 7.5" # set the target architecture manually, suggested in issue https://github.com/NVIDIA/apex/issues/605#issuecomment-554453001
pip3 install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
cd ..
But I got the following error:
The torch. version = 1.8.0 comes with
Compiling cuda extensions with
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11. 1. TC455_06. 29190527 _0
The from/usr/local/cuda - 11.1 / bin

Traceback (most recent call last):
File "", line 1, in
File "/tmp/pip-req-build-u1q04ptr/setup.py", line 177, in
check_cuda_torch_binary_vs_bare_metal(CUDA_HOME)
File "/tmp/pip-req-build-u1q04ptr/setup.py", line 40, in check_cuda_torch_binary_vs_bare_metal

  • "In some cases, a minor-version mismatch will not cause later errors: "
    RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch Pytorch binaries were compiled with Cuda 11.1.
    In some cases, a minor-version mismatch will not cause later errors: https://github.com/NVIDIA/apex/pull/323#discussion_r287021798. You can try commenting out this check (at your own risk).
    Running setup.py install for apex ... error

As shown above, I built pytorch with cuda11.1 and installed cuda11.1. Why did I throw this error?

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 reading setup.py, especially check_cuda_torch_binary_vs_bare_metal at the line shown in the traceback. Reproduce the reported installation with CUDA 11.1, PyTorch 1.8.0, and Ubuntu 20.04, then determine why the check reports a version mismatch and document the verified resolution.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch, ubuntu
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.