実現したいこと
Google Colabでgpuを使用できるようにしたい。
前提
グリッドサーチを行いたいのですが、gpuを使用できません。
ランタイムはgpuに変更しています。
また、gpu使用時間は問題ありません。
該当のソースコード
Python
1# 訓練用データと検証用データに分割2X_train, X_val, y_train, y_val = train_test_split(X, y, test_size=0.2, random_state=42)3 4 5# LightGBMのモデルを定義する6model = lgb.LGBMRegressor()7 8# ハイパーパラメータのグリッドサーチを定義する9param_grid = {10 'num_leaves': [20, 30, 40],11 'learning_rate': [0.05, 0.1, 0.2],12 'n_estimators': [50, 100, 200]13}14 15# グリッドサーチを実行する16grid_search = GridSearchCV(model, param_grid=param_grid, cv=5, n_jobs=-2, verbose=2)17start = time.time()18grid_search.fit(X_train, y_train)19print("グリッドサーチの処理時間:", time.time() - start)20 21# 最適なハイパーパラメータを取得する22best_params = grid_search.best_params_ 23print("最適なハイパーパラメータ:", best_params)24 25# 最適なハイパーパラメータでモデルをトレーニングする26best_model = lgb.LGBMRegressor(**best_params)27best_model.fit(X_train, y_train)28 29# テストデータで予測を行う30y_pred = best_model.predict(X_val)31mse = mean_squared_error(y_val, y_pred)32print("MSE:", mse)
試したこと
gpuのラインタイムに変更。
以下のコードを実装し、gpuの稼働を確認
python
1import tensorflow as tf 2print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))3 4# Num GPUs Available: 1
補足情報(FW/ツールのバージョンなど)
Google Colab 無償版
0 コメント