added pytorch
ci/woodpecker/push/woodpecker Pipeline failed
Details
ci/woodpecker/push/woodpecker Pipeline failed
Details
This commit is contained in:
parent
abb6c34bd6
commit
637c0fb0f7
|
@ -18,6 +18,7 @@ RUN apt update && apt install -y libcudnn8
|
|||
#RUN pip install tensorflow
|
||||
RUN pip install --force-reinstall tf-nightly==2.12.0.dev20230126
|
||||
RUN pip install --upgrade tensorrt
|
||||
RUN pip install pytorch
|
||||
USER jovyan
|
||||
|
||||
# https://anaconda.org/pytorch-test/pytorch-cuda
|
||||
|
|
Loading…
Reference in New Issue