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
else
export FORCE_CUDA=1
case $CUDA_VERSION in
cu112)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.2.*"
;;
cu111)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.1.*"
;;
cu110)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.0.*"
;;
cu102)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==10.2.*"
;;
cu101)
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"
exit 1
......
......@@ -21,7 +21,8 @@ requirements:
build:
string: py{{py}}_torch_{{ environ['TORCH_VERSION'] }}_{{ environ['CUDA_VERSION'] }}
script: pip install .
script: python setup.py install
# script: pip install .
test:
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