pyhtonの仮想環境でのGPU利用について。
windows10でpipを使ってpythonの環境を作っています。anacondaはつかっていません。
venvで仮想環境を作ったあと、必要なライブラリをpip installしました。
その後、pytorchのためのGPU環境をつくるために、CUDAtoolkitをダウンロードして、CUDAをインストールしました。
しかし、仮想環境ではGPUは認識されません。
Python
1print(torch.cuda.is_available()) 2#False
仮想環境を抜けてから、同様の環境をつくるとGPUを認識してくれます。
これはなぜでしょうか。
ライブラリのインストールなどは、仮想環境に入ってから行いましたが、CuDNNのインストールはファイルを
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v.○○
の該当の箇所に配置するだけでしたし、CUDAはデスクトップ上でexeファイルを実行したので、この二つに関しては、仮想環境では何もしていません。
何か仮想環境からしなくてはならない手続きが抜けているのでしょうか。
バージョンなど、仮想環境内の対応関係は間違っていません。
仮想環境を抜けてから作った、GPUが認識された環境と同様です。
GPU RTX 2070 super
GPUドライバ 465.51
python 3.8.8
torch 1.7.1+cu110
CUDA 11.0.3
あなたの回答
tips
プレビュー