search

bytedance / neurst

Neural end-to-end Speech Translation Toolkit
Other
298 stars 45 forks source link

CUDA Version #57

Open HildaNya opened 2 years ago

HildaNya commented 2 years ago

Hi. I'm working on the must-c example and encountered a problem that might be due to CUDA incompatibility. Could you tell me the CUDA version that was used in your setting (I don't think the github page provided this information) ? Also, it seems that my institute's GPUs don't support mixed precision. In this case, should I apply --dtype float32?

Thanks again!

zhaocq-nlp commented 2 years ago

Hi. For GPU support, see https://www.tensorflow.org/install/pip#software_requirements for detailed instructions.

To disable mixed precision computation, set --dtype float32 option.