Open deltaguo opened 12 months ago
is your torch installed by "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.6"?
I am using a container built by DOCKER_BUILDKIT=1 ./docker/build.sh rocm
, which already contains pytorch. The difference from README.md is that I used podman's container. I don't know if this will cause exceptions.
I tried to configure the environment base on docker.io/rocm/pytorch, and the above errors did not occur.
I created the ROCM image according to the instructions and installed AITemplate in the container, but when I executed examples/05_stable_diffusion/compile.py, I was prompted that the nvidia driver was missing?