Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update torch to 1.13.1+cu116 #645

Merged
5 commits merged into from
Jan 28, 2023
Merged

Update torch to 1.13.1+cu116 #645

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
f0d5f5f
Fixes #644; update torch to 1.13.1+cu116
pdmack Jan 27, 2023
7d5b254
Update requirements.txt
pdmack Jan 27, 2023
8248c1a
Update requirements.txt
pdmack Jan 27, 2023
53fbb2e
Merge branch 'branch-23.01' into pdmack_torch-update
pdmack Jan 27, 2023
2c1a844
Fix other missed torch refs
pdmack Jan 27, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions docker/conda/environments/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# Note, to include this when using setup.py or pip, set the variable:
# PIP_FIND_LINKS=https://download.pytorch.org/whl/cu113/torch_stable.html
--find-links https://download.pytorch.org/whl/cu113/torch_stable.html
# PIP_FIND_LINKS=https://download.pytorch.org/whl/cu116/torch_stable.html
--find-links https://download.pytorch.org/whl/cu116/torch_stable.html

####### Pip-only runtime dependencies (keep sorted!) #######
# Packages listed here should also be listed in setup.py
Expand All @@ -10,6 +10,6 @@ jupyter-core>=4.11.2,<5.0
jupyterlab
nvidia-pyindex
pytest-benchmark>=4.0
torch==1.12.0+cu113
torch==1.13.1+cu116
tritonclient[all]==2.17.*
websockets
2 changes: 1 addition & 1 deletion models/training-tuning-scripts/phishing-models/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,6 @@ cupy==10.6.0
numpy==1.22.4
pandas==1.3.5
scikit_learn==1.1.3
torch==1.12.0+cu113
torch==1.13.1+cu116
tqdm==4.64.1
transformers==4.24.0
2 changes: 1 addition & 1 deletion models/training-tuning-scripts/root-cause-models/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,6 @@ cupy==10.6.0
numpy==1.22.4
pandas==1.3.5
scikit_learn==1.1.3
torch==1.12.0+cu113
torch==1.13.1+cu116
tqdm==4.64.1
transformers==4.24.0
2 changes: 1 addition & 1 deletion models/validation-inference-scripts/dfp-models/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,4 +6,4 @@ matplotlib==3.4.2
numpy==1.22.0
pandas==1.0.1
scikit_learn==1.0.2
torch==1.12.0+cu113
torch==1.13.1+cu116
2 changes: 1 addition & 1 deletion models/validation-inference-scripts/phishing-models/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,4 +2,4 @@ cudf==22.8.1
numpy==1.22.4
onnxruntime==1.13.1
scipy==1.9.1
torch==1.12.0+cu113
torch==1.13.1+cu116
2 changes: 1 addition & 1 deletion models/validation-inference-scripts/root-cause-models/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,4 +2,4 @@ cudf==22.8.1
numpy==1.22.4
onnxruntime==1.13.1
scipy==1.9.1
torch==1.12.0+cu113
torch==1.13.1+cu116
4 changes: 2 additions & 2 deletions setup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@

# Required to install torch via setup.py
# Note: this is order dependent
os.environ["PIP_FIND_LINKS"] = "https://download.pytorch.org/whl/cu113/torch_stable.html"
os.environ["PIP_FIND_LINKS"] = "https://download.pytorch.org/whl/cu116/torch_stable.html"

import versioneer # noqa: E402
from setuptools import find_packages # noqa: E402
Expand Down Expand Up @@ -49,7 +49,7 @@
# Only list the packages which cannot be installed via conda here. Should mach the requirements in
# docker/conda/environments/requirements.txt
"dfencoder @ git+https://github.com/nv-morpheus/dfencoder.git@branch-23.01#egg=dfencoder",
"torch==1.12.0+cu113",
"torch==1.13.1+cu116",
"tritonclient[all]==2.17.*", # Force to 2.17 since they require grpcio==1.41 for newer versions
],
license="Apache",
Expand Down