Skip to content
This repository was archived by the owner on Dec 16, 2022. It is now read-only.

Update torch requirement from <1.11.0,>=1.6.0 to >=1.6.0,<1.12.0 #5595

Merged
merged 6 commits into from
Mar 14, 2022
Merged

Update torch requirement from <1.11.0,>=1.6.0 to >=1.6.0,<1.12.0 #5595

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
afec824
Update torch requirement from <1.11.0,>=1.6.0 to >=1.6.0,<1.12.0
dependabot[bot] Mar 11, 2022
65180c2
combine torch-related package upgrades
AkshitaB Mar 12, 2022
27caba5
update ci torch version
AkshitaB Mar 12, 2022
4324b1d
update dockerfile
AkshitaB Mar 12, 2022
531b6d2
invalidate cache
AkshitaB Mar 12, 2022
d68be21
update Makefile and Dockerfile.test
AkshitaB Mar 12, 2022
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
4 changes: 2 additions & 2 deletions .github/workflows/ci.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,11 +18,11 @@ on:

env:
# NOTE: Need to update `TORCH_VERSION`, and `TORCH_*_INSTALL` for new torch releases.
TORCH_VERSION: 1.10.0
TORCH_VERSION: 1.11.0
TORCH_CPU_INSTALL: conda install pytorch torchvision torchaudio cpuonly -c pytorch
TORCH_GPU_INSTALL: conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
# Change this to invalidate existing cache.
CACHE_PREFIX: v10
CACHE_PREFIX: v11
# Disable tokenizers parallelism because this doesn't help, and can cause issues in distributed tests.
TOKENIZERS_PARALLELISM: 'false'
# Disable multithreading with OMP because this can lead to dead-locks in distributed tests.
Expand Down
2 changes: 1 addition & 1 deletion Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# It's built from a wheel installation of allennlp using the base images from
# https://github.com/allenai/docker-images/pkgs/container/pytorch

ARG TORCH=1.10.0-cuda11.3
ARG TORCH=1.11.0-cuda11.3
FROM ghcr.io/allenai/pytorch:${TORCH}

RUN apt-get update && apt-get install -y \
Expand Down
2 changes: 1 addition & 1 deletion Dockerfile.test
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# Used to build an image for running tests.

ARG TORCH=1.10.0-cuda11.3
ARG TORCH=1.11.0-cuda11.3
FROM ghcr.io/allenai/pytorch:${TORCH}

RUN apt-get update && apt-get install -y \
Expand Down
4 changes: 2 additions & 2 deletions Makefile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@ MD_DOCS_TGT = site/
MD_DOCS_EXTRAS = $(addprefix $(MD_DOCS_ROOT),README.md CHANGELOG.md CONTRIBUTING.md)

TORCH_INSTALL = pip install torch torchvision
DOCKER_TORCH_VERSION = 1.10.0-cuda11.3
DOCKER_TEST_TORCH_VERSION = 1.10.0-cuda11.3
DOCKER_TORCH_VERSION = 1.11.0-cuda11.3
DOCKER_TEST_TORCH_VERSION = 1.11.0-cuda11.3

DOCKER_TAG = latest
DOCKER_IMAGE_NAME = allennlp/allennlp:$(DOCKER_TAG)
Expand Down
4 changes: 2 additions & 2 deletions requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
################################
###### Core dependencies #######
################################
torch>=1.6.0,<1.11.0
torchvision>=0.8.1,<0.12.0
torch>=1.6.0,<1.12.0
torchvision>=0.8.1,<0.13.0
cached-path>=1.0.2,<1.2.0
fairscale==0.4.6
jsonnet>=0.10.0 ; sys.platform != 'win32'
Expand Down