Skip to content
Snippets Groups Projects
Commit 815bdd05 authored by Eric Hambro's avatar Eric Hambro
Browse files

Fixup to reflect Heiner's changes to TB.

parent bade7570
No related branches found
No related tags found
1 merge request!3Add Dockerfile for TB baseline.
Hide whitespace changes
Inline Side-by-side
nethack_baselines/torchbeast/Dockerfile
+ 2
− 2
View file @ 815bdd05
# -*- mode: dockerfile -*- # -*- mode: dockerfile -*-
FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04 FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04
ARG PYTHON_VERSION=3.8 ARG PYTHON_VERSION=3.7
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -yq \ RUN apt-get update && apt-get install -yq \
...@@ -43,7 +43,7 @@ RUN python -c 'import torch' ...@@ -43,7 +43,7 @@ RUN python -c 'import torch'
# Install TorchBeast # Install TorchBeast
WORKDIR /opt/ WORKDIR /opt/
RUN git clone https://github.com/condnsdmatters/torchbeast.git --branch eric/experimental-port --recursive RUN git clone https://github.com/facebookresearch/torchbeast.git --recursive
WORKDIR /opt/torchbeast WORKDIR /opt/torchbeast
RUN pip install -r requirements.txt RUN pip install -r requirements.txt
......
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