CLIP-Chinese
CLIP-Chinese copied to clipboard
你好,我遇到了没有cuda的问题
你好,我遇到了没有cuda的问题 在windows后台运行:python train_clip.py --train_args_file train_args/train_clip.json
报错
Traceback (most recent call last):
File "train_clip.py", line 136, in --fp16
) and FP16 half precision evaluation (--fp16_full_eval
) can only be used on CUDA devices.
cuda版本是12.4 (CLIP-Chinese) F:\python\CLIP-Chinese>nvcc -V nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2024 NVIDIA Corporation Built on Tue_Feb_27_16:28:36_Pacific_Standard_Time_2024 Cuda compilation tools, release 12.4, V12.4.99 Build cuda_12.4.r12.4/compiler.33961263_0
是版本不兼容吗?
找到了 配置文件"fp16": 设置成false即可
你没遇到其他的问题吗? 比如: File "/data1/hgc_data/CLIP-Chinese-master/component/datacollator.py", line 21, in call texts.append(data['text']) KeyError: 'text'
@fan1cheng2lin3