Pythonでエラーが出ないが処理が終わりません

前提・実現したいこと

ここで配布されているプログラムを動かしたいと考えています。

以前は動いていたのですが、エラーもなくプログラムが終了しなくなったため途方に暮れている状態です。

動かしたいのはこのテストコードです

python3
1CUDA_VISIBLE_DEVICES=0 python generate.py --ckpt_target ./checkpoints/ffhq_sketches.pt --load_noise noise.pt

試したこと

conda create -n testenv python=3.6 anaconda

で仮想環境を作成後、
必要なptファイルをダウンロードし、checkpointsファルダに移動

pip　install -r requirments

でパッケージのインストール後動かしたいコードを実行しましたが処理が終わらず、nvidia-smiで確認してみてもGPUの使用メモリにも変化がありませんでした。

補足情報（FW/ツールのバージョンなど）

CUDA:　10.1
OS:Ubuntu 20.04.2 LTS

導入パッケージ

Name Version Build Channel
_libgcc_mutex 0.1 main
_openmp_mutex 4.5 1_gnu
ca-certificates 2021.10.26 h06a4308_2
certifi 2020.6.20 pyhd3eb1b0_3
charset-normalizer 2.0.10 pypi_0 pypi
dataclasses 0.8 pypi_0 pypi
future 0.18.2 pypi_0 pypi
idna 3.3 pypi_0 pypi
jsonpatch 1.32 pypi_0 pypi
jsonpointer 2.2 pypi_0 pypi
ld_impl_linux-64 2.35.1 h7274673_9
libffi 3.3 he6710b0_2
libgcc-ng 9.3.0 h5101ec6_17
libgomp 9.3.0 h5101ec6_17
libstdcxx-ng 9.3.0 hd4cf53a_17
lpips 0.1.3 pypi_0 pypi
ncurses 6.3 h7f8727e_2
ninja 1.10.2.3 pypi_0 pypi
numpy 1.19.5 pypi_0 pypi
openssl 1.1.1m h7f8727e_0
pillow 8.0.1 pypi_0 pypi
pip 21.2.2 py36h06a4308_0
python 3.6.13 h12debd9_1
pytorch-fid 0.1.1 pypi_0 pypi
pyzmq 22.3.0 pypi_0 pypi
readline 8.1.2 h7f8727e_1
requests 2.27.1 pypi_0 pypi
scipy 1.5.4 pypi_0 pypi
setuptools 58.0.4 py36h06a4308_0
six 1.16.0 pypi_0 pypi
sqlite 3.37.0 hc218d9a_0
tk 8.6.11 h1ccaba5_0
torch 1.7.0 pypi_0 pypi
torchfile 0.1.0 pypi_0 pypi
torchvision 0.8.1 pypi_0 pypi
tornado 6.1 pypi_0 pypi
tqdm 4.51.0 pypi_0 pypi
typing-extensions 4.0.1 pypi_0 pypi
urllib3 1.26.8 pypi_0 pypi
visdom 0.1.8.9 pypi_0 pypi
websocket-client 1.2.3 pypi_0 pypi
wheel 0.37.1 pyhd3eb1b0_0
xz 5.2.5 h7b6447c_0
zlib 1.2.11 h7f8727e_4