remove thop dependency
This commit is contained in:
parent
ee8988b8a2
commit
11b462e2a9
|
|
@ -2,7 +2,7 @@
|
|||
FROM nvcr.io/nvidia/pytorch:20.03-py3
|
||||
|
||||
# Install dependencies (pip or conda)
|
||||
RUN pip install -U gsutil thop
|
||||
RUN pip install -U gsutil
|
||||
# RUN pip install -U -r requirements.txt
|
||||
|
||||
# Create working directory
|
||||
|
|
|
|||
Loading…
Reference in New Issue