Commit 365b7df7 authored by rusty1s's avatar rusty1s
Browse files

[skip ci] build conda

parent ebc0fe51
...@@ -11,27 +11,15 @@ if [ "${CUDA_VERSION}" = "cpu" ]; then ...@@ -11,27 +11,15 @@ if [ "${CUDA_VERSION}" = "cpu" ]; then
else else
export FORCE_CUDA=1 export FORCE_CUDA=1
case $CUDA_VERSION in case $CUDA_VERSION in
cu112)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.2.*"
;;
cu111) cu111)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.1.*" export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.1.*"
;; ;;
cu110)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.0.*"
;;
cu102) cu102)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==10.2.*" export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==10.2.*"
;; ;;
cu101) cu101)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==10.1.*" export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==10.1.*"
;; ;;
cu100)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==10.0.*"
;;
cu92)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==9.2.*"
;;
*) *)
echo "Unrecognized CUDA_VERSION=$CUDA_VERSION" echo "Unrecognized CUDA_VERSION=$CUDA_VERSION"
exit 1 exit 1
......
...@@ -21,7 +21,8 @@ requirements: ...@@ -21,7 +21,8 @@ requirements:
build: build:
string: py{{py}}_torch_{{ environ['TORCH_VERSION'] }}_{{ environ['CUDA_VERSION'] }} string: py{{py}}_torch_{{ environ['TORCH_VERSION'] }}_{{ environ['CUDA_VERSION'] }}
script: pip install . script: python setup.py install
# script: pip install .
test: test:
imports: imports:
......
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