Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to run in windows or colab pro? #5

Open
bitcoin5000 opened this issue Jun 22, 2020 · 2 comments
Open

How to run in windows or colab pro? #5

bitcoin5000 opened this issue Jun 22, 2020 · 2 comments

Comments

@bitcoin5000
Copy link

bitcoin5000 commented Jun 22, 2020

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
Copy link

I have the same issue

@0xlouis
Copy link

0xlouis commented Sep 30, 2020

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).

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

No branches or pull requests

3 participants