필터 지우기
필터 지우기

Failed to initialize GPU BLAS library

조회 수: 16 (최근 30일)
dhruv khatri
dhruv khatri 2021년 8월 11일
댓글: Liu Xi 2021년 9월 11일
At the start of training my CNN I get the following error:
options = trainingOptions('rmsprop',...
'MaxEpochs', 10,...
'ValidationFrequency', 50,...
'Verbose',true, ...
'InitialLearnRate', 1e-3 ,...
'MiniBatchSize', 64,...
'ExecutionEnvironment', 'gpu',...
'ValidationData', imdsValid_rand, ...
'Shuffle', 'every-epoch', ...
'Plots', 'training-progress',...
'L2Regularization', 0.0010);
%% Train the CNN model
Net_embryo = trainNetwork(imdsTrain_rand, lgraph, options);
Error using trainNetwork (line
184)
Failed to initialize GPU BLAS
library.
Error in TrainCNNClassifier (line
157)
Net_embryo =
trainNetwork(imdsTrain_rand,
lgraph, options);
Caused by:
Error using
gpuArray/hTimesTranspose
Failed to initialize GPU BLAS
library.
Failed to initialize GPU
BLAS library.
My Execution environment is set to 'gpu'. My training progresses smoothly with this set to 'cpu' but it takes a lot of time to train. I have tried updating my NVIDIA driver (latest version)
Output of gpuDevice:
Name: 'NVIDIA GeForce GTX 1660 Ti'
Index: 1
ComputeCapability: '7.5'
SupportsDouble: 1
DriverVersion: 11.4000
ToolkitVersion: 11
MaxThreadsPerBlock: 1024
MaxShmemPerBlock: 49152
MaxThreadBlockSize: [1×3 double]
MaxGridSize: [1×3 double]
SIMDWidth: 32
TotalMemory: 6.4425e+09
AvailableMemory: 4.8578e+09
MultiprocessorCount: 24
ClockRateKHz: 1590000
ComputeMode: 'Default'
GPUOverlapsTransfers: 1
KernelExecutionTimeout: 1
CanMapHostMemory: 1
DeviceSupported: 1
DeviceAvailable: 1
DeviceSelected: 1
nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 471.68 Driver Version: 471.68 CUDA Version: 11.4 |
|-------------------------------+----------------------+----------------------+
| GPU Name TCC/WDDM | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce ... WDDM | 00000000:01:00.0 On | N/A |
| N/A 64C P0 25W / N/A | 1217MiB / 6144MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
I had excatly the same error on my Ubuntu-machine which was fixed after I did sudo apt upgrade. Howerver, I can't seem to find a solution for my Windows system. Problem is mostly a missing library or something. But I can't figure it out. Thank you for help in advance. Best
Dhruv
  댓글 수: 3
dhruv khatri
dhruv khatri 2021년 8월 19일
Thank you Joss, I will try downgrading my drivers.
Liu Xi
Liu Xi 2021년 9월 11일
hello,I meet this problem,too.This may be due to the explosion of gpu memory.Setting parameter 'Batachsize' lower may be able to solve.

댓글을 달려면 로그인하십시오.

답변 (0개)

카테고리

Help CenterFile Exchange에서 Parallel and Cloud에 대해 자세히 알아보기

제품


릴리스

R2021a

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by