This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
ERROR: LoadError: TaskFailedException | |
Stacktrace: | |
[1] wait | |
@ ./task.jl:334 [inlined] | |
[2] fetch(t::Task) | |
@ Base ./task.jl:349 | |
[3] top-level scope | |
@ ~/knightvision/piece_recognition/KnightVisionServer/test/test_cuda.jl:39 | |
nested task error: CUDNNError: CUDNN_STATUS_INTERNAL_ERROR (code 4) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
const LOCK = ReentrantLock() | |
function inference(imgs) | |
lock(LOCK) | |
try | |
out = nn(imgs) | |
return maximum(cpu(out)) | |
finally | |
unlock(LOCK) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
ERROR: LoadError: TaskFailedException | |
Stacktrace: | |
[1] wait | |
@ ./task.jl:334 [inlined] | |
[2] fetch(t::Task) | |
@ Base ./task.jl:349 | |
[3] top-level scope | |
@ ~/knightvision/piece_recognition/KnightVisionServer/test/test_cuda.jl:46 | |
nested task error: Out of GPU memory trying to allocate 13.184 MiB |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
+-----------------------------------------------------------------------------+ | |
| NVIDIA-SMI 470.103.01 Driver Version: 470.103.01 CUDA Version: 11.4 | | |
|-------------------------------+----------------------+----------------------+ | |
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | | |
| | | MIG M. | | |
|===============================+======================+======================| | |
| 0 NVIDIA GeForce ... Off | 00000000:01:00.0 Off | N/A | | |
| 0% 43C P8 10W / 250W | 2256MiB / 11175MiB | 0% Default | | |
| | | N/A | |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
using CUDA | |
import CUDA.CUDNN: cudnnConvolutionForward | |
const W1 = cu(randn(5,5,3,6)) | |
function inference(imgs) | |
out = cudnnConvolutionForward(W1, imgs) | |
return maximum(Array(out)) | |
end |
OlderNewer