Skip to content

Conversation

@Kaszanas
Copy link

As the installation instructions are quite hefty, I have decided to add a Dockerfile. This streamlines the environment setup for anyone using containers.

This was especially helpful for me because installation on Windows failed for me (I am not sure why).

Also I was not sure if you use the GPU capabilities but the image that is built uses PyTorch with CUDA.

There is an alternative to this that may be worth exploring: https://hub.docker.com/r/pytorch/pytorch/tags

This could remove one of the RUN commands:

RUN pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118

nikhilsos pushed a commit to nikhilsos/all-in-one that referenced this pull request Aug 8, 2025
Добавлен план доработок
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants