Replace pytorch location for cuda 11

This commit is contained in:
Andreas Argyriou 2021-11-08 19:44:32 +00:00
Родитель 1d94f13d0f
Коммит 8503c513eb
6 изменённых файлов: 8 добавлений и 8 удалений

Просмотреть файл

@ -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

Просмотреть файл

@ -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"

Просмотреть файл

@ -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"

Просмотреть файл

@ -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"

Просмотреть файл

@ -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"

Просмотреть файл

@ -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
############