Replace pytorch location for cuda 11
This commit is contained in:
Родитель
1d94f13d0f
Коммит
8503c513eb
|
@ -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
|
||||
|
||||
|
||||
############
|
||||
|
|
Загрузка…
Ссылка в новой задаче