pip install pycuda

import pycuda.autoinit
import pycuda.driver as cuda

import pycuda.autoinit
import pycuda.gpuarray as gpuarray

# 定義兩個(gè)向量
a = gpuarray.to_gpu([1, 2, 3, 4])
b = gpuarray.to_gpu([5, 6, 7, 8])

# 執(zhí)行向量加法
c = a + b
print(c)

import numpy as np
import pycuda.autoinit
import pycuda.gpuarray as gpuarray
import pycuda.driver as cuda
from pycuda.elementwise import ElementwiseKernel

# 定義矩陣
A = np.random.randn(3, 3).astype(np.float32)
B = np.random.randn(3, 3).astype(np.float32)

# 將矩陣上傳到 GPU
d_A = cuda.mem_alloc(A.nbytes)
d_B = cuda.mem_alloc(B.nbytes)
cuda.memcpy_htod(d_A, A)
cuda.memcpy_htod(d_B, B)

# 定義矩陣乘法的內(nèi)核函數(shù)
matmul_kernel = ElementwiseKernel(
    "float *A, float *B, float *C",
    "C[i] = A[i] * B[i]",
    "matmul_kernel"
)

# 執(zhí)行矩陣乘法
C = gpuarray.empty_like(A)
matmul_kernel(d_A, d_B, C)

# 從 GPU 獲取結(jié)果
result = np.empty_like(C.get())
cuda.memcpy_dtoh(result, C)

print(result)

import numpy as np
import pycuda.autoinit
import pycuda.gpuarray as gpuarray
import pycuda.driver as cuda
from pycuda.compiler import SourceModule

# 定義 CUDA 核函數(shù)
mod = SourceModule("""
    __global__ void add(int *a, int *b, int *c) {
        int idx = threadIdx.x + blockIdx.x * blockDim.x;
        c[idx] = a[idx] + b[idx];
    }
""")

# 獲取核函數(shù)
add_func = mod.get_function("add")

# 定義輸入數(shù)據(jù)
a = np.array([1, 2, 3, 4]).astype(np.int32)
b = np.array([5, 6, 7, 8]).astype(np.int32)
c = np.zeros_like(a)

# 將數(shù)據(jù)上傳到 GPU
d_a = gpuarray.to_gpu(a)
d_b = gpuarray.to_gpu(b)
d_c = gpuarray.to_gpu(c)

# 執(zhí)行核函數(shù)
block_size = 4
grid_size = len(a) // block_size
add_func(d_a, d_b, d_c, block=(block_size, 1, 1), grid=(grid_size, 1))

# 從 GPU 獲取結(jié)果
result = d_c.get()
print(result)

import numpy as np
import pycuda.autoinit
import pycuda.gpuarray as gpuarray
from pycuda.elementwise import ElementwiseKernel

# 定義矩陣運(yùn)算內(nèi)核函數(shù)
matmul_kernel = ElementwiseKernel(
    "float *A, float *B, float *C",
    "C[i] = A[i] * B[i]",
    "matmul_kernel"
)

# 創(chuàng)建隨機(jī)矩陣
A = np.random.randn(1000, 1000).astype(np.float32)
B = np.random.randn(1000, 1000).astype(np.float32)

# 將矩陣上傳到 GPU
d_A = gpuarray.to_gpu(A)
d_B = gpuarray.to_gpu(B)
d_C = gpuarray.empty_like(d_A)

# 執(zhí)行矩陣運(yùn)算內(nèi)核函數(shù)
matmul_kernel(d_A, d_B, d_C)

# 從 GPU 獲取結(jié)果
result = d_C.get()

print(result)

import numpy as np
import pycuda.autoinit
import pycuda.gpuarray as gpuarray
from sklearn.datasets import make_classification
from sklearn.neural_network import MLPClassifier

# 創(chuàng)建一個(gè)虛擬數(shù)據(jù)集
X, y = make_classification(n_samples=10000, n_features=20, random_state=42)

# 將數(shù)據(jù)上傳到 GPU
d_X = gpuarray.to_gpu(X.astype(np.float32))
d_y = gpuarray.to_gpu(y.astype(np.int32))

# 創(chuàng)建并訓(xùn)練神經(jīng)網(wǎng)絡(luò)模型
mlp = MLPClassifier(hidden_layer_sizes=(100, 50), max_iter=100)
mlp.fit(d_X.get(), d_y.get())

# 打印模型評(píng)分
score = mlp.score(d_X.get(), d_y.get())
print("模型評(píng)分:", score)

import numpy as np
import pycuda.autoinit
import pycuda.gpuarray as gpuarray
from tensorflow.keras.datasets import mnist
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
from tensorflow.keras.utils import to_categorical

# 加載 MNIST 數(shù)據(jù)集
(X_train, y_train), (X_test, y_test) = mnist.load_data()

# 數(shù)據(jù)預(yù)處理
X_train = X_train.reshape(-1, 784).astype('float32') / 255.0
X_test = X_test.reshape(-1, 784).astype('float32') / 255.0
y_train = to_categorical(y_train, 10)
y_test = to_categorical(y_test, 10)

# 將數(shù)據(jù)上傳到 GPU
d_X_train = gpuarray.to_gpu(X_train)
d_y_train = gpuarray.to_gpu(y_train)
d_X_test = gpuarray.to_gpu(X_test)
d_y_test = gpuarray.to_gpu(y_test)

# 創(chuàng)建并訓(xùn)練深度學(xué)習(xí)模型
model = Sequential([
    Dense(512, activation='relu', input_shape=(784,)),
    Dense(256, activation='relu'),
    Dense(10, activation='softmax')
])
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
model.fit(d_X_train.get(), d_y_train.get(), epochs=10, batch_size=128)

# 評(píng)估模型
loss, accuracy = model.evaluate(d_X_test.get(), d_y_test.get())
print('測(cè)試集損失:', loss)
print('測(cè)試集準(zhǔn)確率:', accuracy)