mirror of
https://github.com/CodedotAl/gpt-code-clippy.git
synced 2024-08-17 02:40:33 +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
|
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