Commit 49af838c authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

torch 1.6.0

parent 3e0896a7
......@@ -58,4 +58,4 @@ RUN wget -nv -O miniconda.sh https://repo.anaconda.com/miniconda/Miniconda3-py37
ENV PATH ${CONDA_DIR}/bin:${PATH}
RUN pip install -r requirements.txt --no-cache-dir
RUN pip install -r requirements.txt --no-cache-dir -f https://download.pytorch.org/whl/torch_stable.html
ray[rllib]==0.8.5
procgen==0.10.1
torch==1.3.1
torchvision==0.4.2
ray[rllib]==0.8.6
procgen==0.10.4
torch==1.6.0+cu101
torchvision==0.7.0+cu101
tensorflow==2.2.0
mlflow==1.8.0
boto3==1.13.10
scikit-image
requests
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment