Merge pull request #331 from Visual-Behavior/torch_v2

Torch v2

Author: thibo73800

Dockerfile

@@ -4,10 +4,10 @@ FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
 #FROM nvidia/cuda:11.6.0-cudnn8-devel-ubuntu20.04
 
 ARG py=3.9
-ARG pytorch=1.13.1
-ARG torchvision=0.14.1
-ARG torchaudio=0.13.1
-ARG pytorch_lightning=1.9.0
+ARG pytorch=2.1.0.dev20230313+cu117
+ARG torchvision=0.15.0.dev20230313+cu117
+ARG torchaudio=2.0.0.dev20230313+cu117
+ARG pytorch_lightning=1.9.3
 ARG pycyda=11.7
 
 ENV TZ=Europe/Paris
@@ -30,8 +30,9 @@ ENV HOME /workspace
 WORKDIR /workspace
 
 # Pytorch & pytorch litning
-RUN conda install pytorch==${pytorch} torchvision==${torchvision} torchaudio==${torchaudio} pytorch-cuda=${pycuda} -c pytorch -c nvidia
+#RUN conda install py pytorch-cuda=${pycuda} -c pytorch -c nvidia
+RUN pip install --pre torch==${pytorch} torchvision==${torchvision} torchaudio==${torchaudio} --index-url https://download.pytorch.org/whl/nightly/cu117
 RUN pip install pytorch_lightning==${pytorch_lightning}
 
-COPY requirements-torch1.13.1.txt /install/requirements-torch1.13.1.txt
-RUN pip install -r /install/requirements-torch1.13.1.txt
+COPY requirements-torch2.1.txt /install/requirements-torch2.1.txt
+RUN pip install -r /install/requirements-torch2.1.txt

README.md

@@ -4,7 +4,7 @@
 
 <a href="https://visual-behavior.github.io/aloception-oss/">Documentation</a>
 
-[![Conventional Commits](https://img.shields.io/badge/Conventional%20Commits-0.5.0-green.svg)](https://conventionalcommits.org)
+[![Conventional Commits](https://img.shields.io/badge/Conventional%20Commits-0.5.1-green.svg)](https://conventionalcommits.org)
 
 # Aloception open source software
 
@@ -78,17 +78,17 @@ training pipelines with **augmented tensors**.
 ### Docker install
 
 ```
-docker build -t aloception-oss:cuda-11.3.1-pytorch1.13.1-lightning1.9.0 .
+docker build -t aloception-oss:cuda-11.7-pytorch2.1.0-lightning1.9.3 .
 ```
 
 ```
-docker run --gpus all -it -v /YOUR/WORKSPACE/:/workspace --privileged -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix aloception-oss:cuda-11.3.1-pytorch1.13.1-lightning1.9.0
+docker run --gpus all -it -v /YOUR/WORKSPACE/:/workspace --privileged -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix aloception-oss:cuda-11.7-pytorch2.1.0-lightning1.9.3
 ```
 
 Or without building the image
 
 ```
-docker run --gpus all -it -v /YOUR/WORKSPACE/:/workspace --privileged -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix visualbehaviorofficial/aloception-oss:cuda-11.3.1-pytorch1.13.1-lightning1.9.0
+docker run --gpus all -it -v /YOUR/WORKSPACE/:/workspace --privileged -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix visualbehaviorofficial/aloception-oss:cuda-11.7-pytorch2.1.0-lightning1.9.3
 ```
 
 

requirements-torch2.1.txt

@@ -0,0 +1,23 @@
+pycocotools==2.0.2
+PyYAML==5.4.1
+chardet==4.0.0
+idna==2.10
+
+scipy==1.10.0
+
+more_itertools==8.8.0
+requests==2.25.1
+opencv-python==4.7.0.68
+
+python-dateutil==2.8.2
+urllib3==1.26.6
+
+protobuf==4.21.12
+wandb==0.13.9
+
+tqdm==4.62.3
+captum==0.4.0
+
+setuptools==59.5.0
+
+numpy==1.23.5