More than 1 year has passed since last update.

VScodeでGPUが認識されず使えない（colaboratoryのipynbファイル）

Last updated at 2022-07-13Posted at 2022-06-26

colaboratoryのipynbファイルを、ローカルPCのVScodeで実行したらGPUが認識されず使えない

・新規でipynbファイルを作成したことで解決
・理由はわからない。
・colaboratoryからダウンロードしたipynbファイルになんらかの情報がのこっているのかな。

colaboratoryで学習をまわしていたが頻繁にセッション切断されるため、ローカルPCのgpuで学習させようと思いました。
colaboratoryからipynbファイルをダウンロードしてVScodeに取り込んで実行しましたが、GPUが利用できなかった。

GPUはRTX3080laptop
CUDA11.7、cuDNNv8の最新

VSCode上でvenvで仮想環境を作り、Pytorchのバージョンは1.11 cuda
https://pytorch.org/

colaboratoryからダウンロードしたipynbファイルで実行すると

import torch
torch.__version__,torch.cuda.is_available()

結果は、('1.11.0+cpu', False)

新しくipynbファイルを作成し実行すると

結果は、('1.11.0+cu113', True)

どうゆうこっちゃ。