test torch 2.0.1 (#741)

Signed-off-by: Clement Fuji Tsang <cfujitsang@nvidia.com>

ci/gitlab_jenkins_templates/core_ci.jenkins

@@ -5,17 +5,18 @@ import groovy.transform.Field
 if (gitlabActionType == "MERGE" || gitlabSourceBranch == "master") {
 gitlabCommitStatus("launch all builds") {
 
+
 // Configs for build from pytorch docker images
 // (See: https://hub.docker.com/r/pytorch/pytorch/tags)
 def ubuntu_from_pytorch_configs = [
     [
         // python: 3.7
-        'cudaVer': '11.3', 'cudnnVer': '8', 'torchVer': '1.12.1',
+        'cudaVer': '11.6', 'cudnnVer': '8', 'torchVer': '1.13.1',
         'archsToTest': 'MULTI'
     ],
     [
         // python: 3.10
-        'cudaVer': '11.6', 'cudnnVer': '8', 'torchVer': '1.13.1',
+        'cudaVer': '11.7', 'cudnnVer': '8', 'torchVer': '2.0.1',
         'archsToTest': 'MULTI'
     ]
 ]
@@ -34,13 +35,13 @@ def ubuntu_from_nvcr_configs = [
 // (See: https://hub.docker.com/r/nvidia/cuda/tags)
 def ubuntu_from_cuda_configs = [
     [
-        'cudaVer': '11.3.0', 'cudnnVer': '8',
-        'pythonVer': '3.8', 'torchVer': '1.10.2',
+        'cudaVer': '11.3.1', 'cudnnVer': '8',
+        'pythonVer': '3.8', 'torchVer': '1.11.0',
         'archsToTest': 'MULTI'
     ],
     [
         'cudaVer': '11.8.0', 'cudnnVer': '8',
-        'pythonVer': '3.10', 'torchVer': '2.0.0',
+        'pythonVer': '3.10', 'torchVer': '2.0.1',
         'archsToTest': 'MULTI'
     ],
 ]
@@ -49,10 +50,10 @@ def ubuntu_from_cuda_configs = [
 // (Use docker image ubuntu:18.04 as a base)
 def ubuntu_cpuonly_configs = [
     [
-        'pythonVer': '3.8', 'torchVer': '1.10.2',
+        'pythonVer': '3.8', 'torchVer': '1.11.0',
     ],
     [
-        'pythonVer': '3.9', 'torchVer': '1.13.1',
+        'pythonVer': '3.10', 'torchVer': '2.0.1',
     ]
 ]
 
@@ -232,12 +233,12 @@ def windows_from_server_configs = [
     //      two machines exist, only the TITAN RTX will pass tests
     [
         'cudaVer': '11.3',
-        'pythonVer': '3.8', 'torchVer': '1.10.2',
+        'pythonVer': '3.8', 'torchVer': '1.11.0',
         'archsToTest': 'gpu_tester' //'Tesla_V100_PCIE_32GB'
     ],
     [
-        'cudaVer': '11.6',
-        'pythonVer': '3.9', 'torchVer': '1.13.1',
+        'cudaVer': '11.8',
+        'pythonVer': '3.10', 'torchVer': '2.0.1',
         'archsToTest': 'gpu_tester' //'Tesla_V100_PCIE_32GB'
     ]
 ]

ci/gitlab_jenkins_templates/windows_build_CI.jenkins

@@ -10,7 +10,8 @@ def cuda_version_url = [
     '11.3': 'https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.89_win10.exe',
     '11.5': 'https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_496.13_windows.exe',
     '11.6': 'https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_511.65_windows.exe',
-    '11.7': 'https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_516.94_windows.exe'
+    '11.7': 'https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda_11.7.1_516.94_windows.exe',
+    '11.8': 'https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_522.06_windows.exe'
 ]
 
 

examples/tutorial/tutorial_common.py

@@ -16,7 +16,6 @@
 import math
 import os
 import torch
-import torchvision
 
 try:
     import matplotlib.pyplot as plt

setup.py

@@ -14,7 +14,7 @@
 import warnings
 
 TORCH_MIN_VER = '1.6.0'
-TORCH_MAX_VER = '2.0.0'
+TORCH_MAX_VER = '2.0.1'
 CYTHON_MIN_VER = '0.29.20'
 IGNORE_TORCH_VER = os.getenv('IGNORE_TORCH_VER') is not None
 

tools/linux/Dockerfile.install

@@ -79,5 +79,4 @@ RUN cd /tmp && \
     cp ./docker/10_nvidia.json /usr/share/glvnd/egl_vendor.d/10_nvidia.json && \
     pip install .
 
-ENV IGNORE_TORCH_VER "1"
 RUN python setup.py develop

tools/requirements.txt

@@ -1,4 +1,4 @@
-usd-core<=23.5
+usd-core<=23.5 ; python_version < "3.11"
 numpy
 pybind11
 Pillow>=8.0.0

tools/windows/Dockerfile

@@ -111,7 +111,6 @@ RUN pip install --no-cache-dir --trusted-host pypi.org --trusted-host pypi.pytho
     --trusted-host files.pythonhosted.org -r tools\ci_requirements.txt \
     -r tools\viz_requirements.txt -r tools\requirements.txt -r tools/build_requirements.txt
 
-ENV IGNORE_TORCH_VER "1"
 ENV TORCH_CUDA_ARCH_LIST "7.0 7.5+PTX"
 RUN python setup.py develop