From 8503c513ebd6bc093572030d8916b2d1d6e1497c Mon Sep 17 00:00:00 2001 From: Andreas Argyriou Date: Mon, 8 Nov 2021 19:44:32 +0000 Subject: [PATCH] Replace pytorch location for cuda 11 --- recommenders/README.md | 2 +- tests/ci/azure_pipeline_test/dsvm_nightly_linux_gpu.yml | 2 +- tests/ci/azure_pipeline_test/dsvm_notebook_linux_gpu.yml | 2 +- tests/ci/azure_pipeline_test/dsvm_unit_linux_gpu.yml | 2 +- tests/ci/azure_pipeline_test/release_pipeline.yml | 6 +++--- tools/docker/Dockerfile | 2 +- 6 files changed, 8 insertions(+), 8 deletions(-) diff --git a/recommenders/README.md b/recommenders/README.md index 0852c350..5ab17b2b 100644 --- a/recommenders/README.md +++ b/recommenders/README.md @@ -57,7 +57,7 @@ For manual installation of the necessary requirements see [TensorFlow](https://w When installing with GPU support you will need to point to the PyTorch index to ensure you are downloading a version of PyTorch compiled with CUDA support. This can be done using the --find-links or -f option below. -`pip install --no-cache --no-binary scikit-surprise recommenders[gpu] -f https://download.pytorch.org/whl/cu100/torch_stable.html` +`pip install --no-cache --no-binary scikit-surprise recommenders[gpu] -f https://download.pytorch.org/whl/cu111/torch_stable.html` ## Experimental dependencies diff --git a/tests/ci/azure_pipeline_test/dsvm_nightly_linux_gpu.yml b/tests/ci/azure_pipeline_test/dsvm_nightly_linux_gpu.yml index ef26fb54..7e8f59ef 100644 --- a/tests/ci/azure_pipeline_test/dsvm_nightly_linux_gpu.yml +++ b/tests/ci/azure_pipeline_test/dsvm_nightly_linux_gpu.yml @@ -32,6 +32,6 @@ extends: timeout: 240 conda_env: "nightly_linux_gpu" conda_opts: "python=3.6 -c conda-forge cudatoolkit=11.2 cudnn=8.1" - pip_opts: "[gpu,examples,dev] -f https://download.pytorch.org/whl/cu100/torch_stable.html" + pip_opts: "[gpu,examples,dev] -f https://download.pytorch.org/whl/cu111/torch_stable.html" pytest_markers: "not spark and gpu" pytest_params: "-x" diff --git a/tests/ci/azure_pipeline_test/dsvm_notebook_linux_gpu.yml b/tests/ci/azure_pipeline_test/dsvm_notebook_linux_gpu.yml index 2029e538..d62bf517 100644 --- a/tests/ci/azure_pipeline_test/dsvm_notebook_linux_gpu.yml +++ b/tests/ci/azure_pipeline_test/dsvm_notebook_linux_gpu.yml @@ -60,5 +60,5 @@ extends: task_name: "Test - Unit Notebook Linux GPU" conda_env: "unit_notebook_linux_gpu" conda_opts: "python=3.6 -c conda-forge cudatoolkit=11.2 cudnn=8.1" - pip_opts: "[gpu,examples,dev] -f https://download.pytorch.org/whl/cu100/torch_stable.html" + pip_opts: "[gpu,examples,dev] -f https://download.pytorch.org/whl/cu111/torch_stable.html" pytest_markers: "notebooks and not spark and gpu" diff --git a/tests/ci/azure_pipeline_test/dsvm_unit_linux_gpu.yml b/tests/ci/azure_pipeline_test/dsvm_unit_linux_gpu.yml index f8ace025..70f51cc6 100644 --- a/tests/ci/azure_pipeline_test/dsvm_unit_linux_gpu.yml +++ b/tests/ci/azure_pipeline_test/dsvm_unit_linux_gpu.yml @@ -60,5 +60,5 @@ extends: task_name: "Test - Unit Linux GPU" conda_env: "unit_linux_gpu" conda_opts: "python=3.6 -c conda-forge cudatoolkit=11.2 cudnn=8.1" - pip_opts: "[gpu,dev] -f https://download.pytorch.org/whl/cu100/torch_stable.html" + pip_opts: "[gpu,dev] -f https://download.pytorch.org/whl/cu111/torch_stable.html" pytest_markers: "not notebooks and not spark and gpu" diff --git a/tests/ci/azure_pipeline_test/release_pipeline.yml b/tests/ci/azure_pipeline_test/release_pipeline.yml index 874b8126..e5ebbc48 100644 --- a/tests/ci/azure_pipeline_test/release_pipeline.yml +++ b/tests/ci/azure_pipeline_test/release_pipeline.yml @@ -46,7 +46,7 @@ jobs: task_name: "Test - Unit Linux GPU" conda_env: "release_unit_linux_gpu" conda_opts: "python=3.6 -c conda-forge cudatoolkit=11.2 cudnn=8.1" - pip_opts: "[gpu] -f https://download.pytorch.org/whl/cu100/torch_stable.html" + pip_opts: "[gpu] -f https://download.pytorch.org/whl/cu111/torch_stable.html" pytest_markers: "not notebooks and not spark and gpu" install: "release" @@ -57,7 +57,7 @@ jobs: task_name: "Test - Unit Notebook Linux GPU" conda_env: "release_unit_notebook_linux_gpu" conda_opts: "python=3.6 -c conda-forge cudatoolkit=11.2 cudnn=8.1" - pip_opts: "[gpu,examples] -f https://download.pytorch.org/whl/cu100/torch_stable.html" + pip_opts: "[gpu,examples] -f https://download.pytorch.org/whl/cu111/torch_stable.html" pytest_markers: "notebooks and not spark and gpu" install: "release" @@ -106,7 +106,7 @@ jobs: timeout: 240 conda_env: "release_nightly_linux_gpu" conda_opts: "python=3.6 -c conda-forge cudatoolkit=11.2 cudnn=8.1" - pip_opts: "[gpu,examples] -f https://download.pytorch.org/whl/cu100/torch_stable.html" + pip_opts: "[gpu,examples] -f https://download.pytorch.org/whl/cu111/torch_stable.html" pytest_markers: "not spark and gpu" install: "release" diff --git a/tools/docker/Dockerfile b/tools/docker/Dockerfile index 1a2a29f5..3d7f727c 100644 --- a/tools/docker/Dockerfile +++ b/tools/docker/Dockerfile @@ -137,7 +137,7 @@ RUN if [ "${VIRTUAL_ENV}" = "virtualenv" ] ; then python3.7 -m virtualenv $HOME/ pip install --no-cache --no-binary scikit-surprise recommenders[gpu,xlearn,examples]; fi RUN if [ "${VIRTUAL_ENV}" = "conda" ] ; then \ - pip install --no-cache --no-binary scikit-surprise recommenders[gpu,xlearn,examples] -f https://download.pytorch.org/whl/cu100/torch_stable.html ; fi + pip install --no-cache --no-binary scikit-surprise recommenders[gpu,xlearn,examples] -f https://download.pytorch.org/whl/cu111/torch_stable.html ; fi ############