mirror of
https://github.com/CodedotAl/gpt-code-clippy.git
synced 2024-07-14 18:20:24 +03:00
Update Dockerfile a bit
This commit is contained in:
parent
e3cc5fee4e
commit
c433efeadb
@ -1 +1,6 @@
|
||||
from tensorflow/tensorflow:2.5.0-gpu
|
||||
|
||||
RUN apt update && apt install git vim python3.8 -y
|
||||
RUN pip install datasets \
|
||||
git+https://github.com/huggingface/transformers \
|
||||
torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
|
Loading…
Reference in New Issue
Block a user