Vincent C

icon location Focus on Machine Learning, Data Analysis, CV and NLP.

Results 4 comments of Vincent C

Trained a transformer model ,but cannot infer (CUDA ERROR): invalid device function

my configuration: lightseq 2.1.4 ninja 1.10.2.1 torch 1.7.1 torchaudio 0.7.0a0+a853dff torchvision 0.8.2

Trained a transformer model ,but cannot infer (CUDA ERROR): invalid device function

+-----------------------------------------------------------------------------+ | NVIDIA-SMI 418.87.01 Driver Version: 418.87.01 CUDA Version: 10.1 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage |...

Trained a transformer model ,but cannot infer (CUDA ERROR): invalid device function

hello @Taka152 thank you for your reply how could i build from source? do you mean that git clone first, make the change, then do "pip install -e ."? thank...

cuda, torch version

hello @Jack47 , thank you for your reply may i ask your cuda driver version? thank you!