diff --git a/packaging/conda/variants/cu102.yaml b/packaging/conda/variants/cu102.yaml deleted file mode 100644 index 575f522..0000000 --- a/packaging/conda/variants/cu102.yaml +++ /dev/null @@ -1,10 +0,0 @@ -cuda: - - 10.2 # [linux64] -cuda_home: - - /usr/local/cuda-10.2 # [linux64] -cudnn: - - 7.6.5 # [linux64] -cxx_compiler_version: - - 8.5.0 # [linux64] -pytorch_variant: - - cuda # [linux64] diff --git a/packaging/conda/variants/cu113.yaml b/packaging/conda/variants/cu113.yaml deleted file mode 100644 index ed55ce1..0000000 --- a/packaging/conda/variants/cu113.yaml +++ /dev/null @@ -1,10 +0,0 @@ -cuda: - - 11.3 # [linux64] -cuda_home: - - /usr/local/cuda-11.3 # [linux64] -cudnn: - - 8.2.0 # [linux64] -cxx_compiler_version: - - 9.4.0 # [linux64] -pytorch_variant: - - cuda # [linux64] diff --git a/packaging/conda/variants/cu116.yaml b/packaging/conda/variants/cu117.yaml similarity index 69% rename from packaging/conda/variants/cu116.yaml rename to packaging/conda/variants/cu117.yaml index fed4ec3..0a44fad 100644 --- a/packaging/conda/variants/cu116.yaml +++ b/packaging/conda/variants/cu117.yaml @@ -1,7 +1,7 @@ cuda: - - 11.6 # [linux64] + - 11.7 # [linux64] cuda_home: - - /usr/local/cuda-11.6 # [linux64] + - /usr/local/cuda-11.7 # [linux64] cudnn: - 8.3.2 # [linux64] cxx_compiler_version: diff --git a/packaging/conda/variants/cu118.yaml b/packaging/conda/variants/cu118.yaml new file mode 100644 index 0000000..6378305 --- /dev/null +++ b/packaging/conda/variants/cu118.yaml @@ -0,0 +1,10 @@ +cuda: + - 11.8 # [linux64] +cuda_home: + - /usr/local/cuda-11.8 # [linux64] +cudnn: + - 8.3.2 # [linux64] +cxx_compiler_version: + - 11.2.0 # [linux64] +pytorch_variant: + - cuda # [linux64]