Skip to content

Instantly share code, notes, and snippets.

@hkboujrida
Last active March 4, 2024 19:47
Show Gist options
  • Save hkboujrida/9395133560dd5a43bc99ce82b15284c7 to your computer and use it in GitHub Desktop.
Save hkboujrida/9395133560dd5a43bc99ce82b15284c7 to your computer and use it in GitHub Desktop.
docker run -it \
--name pytorch \
--network=host \
--cap-add=SYS_PTRACE \
--security-opt seccomp=unconfined \
--device=/dev/kfd \
--device=/dev/dri \
--group-add video \
--ipc=host \
--shm-size 8G \
-v $HOME/dockerx:/dockerx \
rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2
GitPython
Pillow
accelerate
blendmodes
clean-fid
einops
facexlib
fastapi==0.90.1
gradio==3.41.2
inflection
jsonmerge
kornia
lark
numpy
omegaconf
open-clip-torch==2.20.0
piexif
psutil
pytorch_lightning==1.6.5
requests
resize-right
safetensors
scikit-image>=0.19
tomesd
torch
torchdiffeq
torchsde
transformers==4.30.2
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment