We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d504261 commit b1c20aaCopy full SHA for b1c20aa
tox.ini
@@ -39,7 +39,7 @@ deps =
39
commands =
40
# if CUDA is available, install the appropriate cupy version
41
bash -c 'if command -v nvcc >/dev/null 2>&1; then \
42
- CUDA_MAJOR=$$(python - <<'"'"'PY'"'"'\nimport re\nimport subprocess\noutput = subprocess.check_output(["nvcc", "--version"], text=True)\nmatch = re.search(r"release\\s+(\\d+)\\.", output)\nprint(match.group(1) if match else "")\nPY\n); \
+ CUDA_MAJOR=$$(nvcc --version | sed -n "s/.*release \([0-9][0-9]*\)\..*/\1/p" | head -n 1); \
43
if [ "$$CUDA_MAJOR" = "12" ]; then \
44
python -m pip install -U cupy-cuda12x; \
45
elif [ "$$CUDA_MAJOR" = "13" ]; then \
0 commit comments