Skip to content

GitLab

Commit ff5208db authored by rusty1s's avatar rusty1s
Browse files

[ci skip]

parent e44831b0
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
conda/pytorch-spline-conv/build_conda.sh
View file @ ff5208db
...@@ -10,6 +10,9 @@ if [ "${CUDA_VERSION}" = "cpu" ]; then ...@@ -10,6 +10,9 @@ if [ "${CUDA_VERSION}" = "cpu" ]; then
export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly # [not osx]"
else else
case $CUDA_VERSION in case $CUDA_VERSION in
cu113)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.3.*"
;;
cu111) cu111)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.1.*" export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.1.*"
;; ;;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment