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
export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly # [not osx]"
else
case $CUDA_VERSION in
cu113)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.3.*"
;;
cu111)
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