Giter VIP home page Giter VIP logo

Comments (6)

albertfgu avatar albertfgu commented on August 27, 2024

I don't have enough information. What OS are you using?

from mamba.

getorca avatar getorca commented on August 27, 2024

post the full stack trace. But kinda looks like you're missing nvcc

from mamba.

alexanderwerning avatar alexanderwerning commented on August 27, 2024

I had the same issue, bare_metal_version is only defined in the setup.py if the CUDA_HOME environment variable is set, but the 11.8 version check is outside of that if block:

mamba/setup.py

Lines 100 to 113 in 009bec5

if CUDA_HOME is not None:
_, bare_metal_version = get_cuda_bare_metal_version(CUDA_HOME)
if bare_metal_version < Version("11.6"):
raise RuntimeError(
f"{PACKAGE_NAME} is only supported on CUDA 11.6 and above. "
"Note: make sure nvcc has a supported version by running nvcc -V."
)
cc_flag.append("-gencode")
cc_flag.append("arch=compute_70,code=sm_70")
cc_flag.append("-gencode")
cc_flag.append("arch=compute_80,code=sm_80")
if bare_metal_version >= Version("11.8"):
cc_flag.append("-gencode")

CUDA 11.8 was installed on my machine, so once I set the CUDA_HOME variable I could install the package.
Maybe an error message about CUDA_HOME would be helpful?

from mamba.

Rhys-McAlister avatar Rhys-McAlister commented on August 27, 2024

I had the same issue, bare_metal_version is only defined in the setup.py if the CUDA_HOME environment variable is set, but the 11.8 version check is outside of that if block:

mamba/setup.py

Lines 100 to 113 in 009bec5

if CUDA_HOME is not None:
_, bare_metal_version = get_cuda_bare_metal_version(CUDA_HOME)
if bare_metal_version < Version("11.6"):
raise RuntimeError(
f"{PACKAGE_NAME} is only supported on CUDA 11.6 and above. "
"Note: make sure nvcc has a supported version by running nvcc -V."
)
cc_flag.append("-gencode")
cc_flag.append("arch=compute_70,code=sm_70")
cc_flag.append("-gencode")
cc_flag.append("arch=compute_80,code=sm_80")
if bare_metal_version >= Version("11.8"):
cc_flag.append("-gencode")

CUDA 11.8 was installed on my machine, so once I set the CUDA_HOME variable I could install the package. Maybe an error message about CUDA_HOME would be helpful?

how do you set the home variable? I'm having this same issue. Thanks in advance

from mamba.

alexanderwerning avatar alexanderwerning commented on August 27, 2024

It depends on the location and version of your cuda installation, for me it was this:
export CUDA_HOME=/usr/local/cuda-11.8

from mamba.

lxaw avatar lxaw commented on August 27, 2024

Has this been solved? Thanks!

from mamba.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.