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 tensorflow
|
||||||
RUN pip install --force-reinstall tf-nightly==2.12.0.dev20230126
|
RUN pip install --force-reinstall tf-nightly==2.12.0.dev20230126
|
||||||
RUN pip install --upgrade tensorrt
|
RUN pip install --upgrade tensorrt
|
||||||
|
RUN pip install pytorch
|
||||||
USER jovyan
|
USER jovyan
|
||||||
|
|
||||||
# https://anaconda.org/pytorch-test/pytorch-cuda
|
# https://anaconda.org/pytorch-test/pytorch-cuda
|
||||||
|
|
Loading…
Reference in New Issue