In[]:=
Needs["CUDALink`"]​​CUDAInformation[]
Out[]=
{1{NameNVIDIA GeForce RTX 3090,Clock Rate1695000,Compute Capabilities8.6,GPU Overlap1,Maximum Block Dimensions{1024,1024,64},Maximum Grid Dimensions{2147483647,65535,65535},Maximum Threads Per Block1024,Maximum Shared Memory Per Block49152,Total Constant Memory65536,Warp Size32,Maximum Pitch2147483647,Maximum Registers Per Block65536,Texture Alignment512,Multiprocessor Count82,Core CountIndeterminate,Execution Timeout1,IntegratedFalse,Can Map Host MemoryTrue,Compute ModeDefault,Texture1D Width131072,Texture2D Width131072,Texture2D Height65536,Texture3D Width16384,Texture3D Height16384,Texture3D Depth16384,Texture2D Array Width32768,Texture2D Array Height32768,Texture2D Array Slices2048,Surface Alignment512,Concurrent KernelsTrue,ECC EnabledFalse,TCC EnabledFalse,Total Memory25769803776}}
In[]:=
InstallCUDA[]
Out[]=
Success
✓
Message: CUDALink installation complete.
CUDAVersion: 11.3

In[]:=
CUDAQ[]
Out[]=
True
In[]:=
CUDAFoldList[#,0,Range[10]]&/@{Plus,Minus,Min,Max,Times}
Out[]=
{{0,1,3,6,10,15,21,28,36,45},{0,-1,1,4,8,13,19,26,34,43},{0,0,0,0,0,0,0,0,0,0},{0,1,2,3,4,5,6,7,8,9},{0,0,0,0,0,0,0,0,0,0}}
In[]:=
TrainingData=RandomReal[1,{10000,4}]RandomReal[1,{10000,4}];
In[]:=
net=NetChain[{8,4}]
Out[]=
NetChain
uniniti
aliz
ed
Inputport:
array
Outputport:
vector(size: 4)

In[]:=
t=AbsoluteTime[];​​NetTrain[net,TrainingData,BatchSize10000,TargetDevice"GPU"];​​Print[AbsoluteTime[]-t];
11.6887363