openai/image-gpt

How to run in windows or colab pro?

bitcoin5000 opened this issue · 2 comments

bitcoin5000 commented

Do i need install linux, or it can be runned in windows env?
Also, can it be run in 2x gtx 1080 ?
And what averange speed of generation will be for one frame?

i try run in windows and get error:
line 1801, in init
self._traceback = tf_stack.extract_stack()

DataLossError (see above for traceback): Unable to open table file :\image-gpt\image-gpt-master\download: Unknown: NewRandomAccessFile failed to Create/Open: \image-gpt-master\download : Access is denied.
; Input/output error
[[node save/RestoreV2 (defined at src/run.py:179) ]]
[[node save/RestoreV2 (defined at src/run.py:179) ]]

zendevil commented

I have the same issue

0xlouis commented

You must specify the name of the template, not the directory.

In my case :
python src/run.py --sample --n_embd 512 --n_head 8 --n_layer 24 --n_gpu 1 --ckpt_path checkpoints\model.ckpt-1000000 --data_path dataset\imagenet --color_cluster_path color\kmeans_centers.npy --save_dir out

Furthermore, if you get a memory error, you can add this to the beginning of the run.py :
os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"

i-GPT2 work fine on single GPU (RTX2080TI) on Windows 10 (~3it/s).

  • 👍1