Skip to content
This repository has been archived by the owner on Jan 27, 2024. It is now read-only.

The translation process cannot use the GPU #5

Closed
ZhengWeiH opened this issue Dec 10, 2019 · 0 comments
Closed

The translation process cannot use the GPU #5

ZhengWeiH opened this issue Dec 10, 2019 · 0 comments

Comments

@ZhengWeiH
Copy link

Although I use the command ： CUDA_VISIBLE_DEVICES=0 python translation.py --config translation.yml, the translation process is always performed on the CPU, how can I solve this problem?
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@ZhengWeiH