pytorchを実行するときにメモリメモリが不足するCUDAの問題を解決する

pytorch実行時のメモリ不足の問題を解決

バージョン:
python:3.7
pytorch:1.2
cuda:10.1

pytorch移行学習中に不十分なビデオメモリが発生しました

それは

nvidia-smiを使用してgpu情報を表示します(最初にC:\ Program Files \ NVIDIA Corporation \ NVSMIをパスに追加する必要があります)

実際に私のビデオメモリは十分であることがわかりました

tensorflowはpytorchと競合する可能性があると言われています

tensorflow-gpuをアンインストールしてみてください

無効です

プログラムにコードを入力してください

import os
os.environ['CUDA_VISIBLE_DEVICES]='23'

プログラムは正常に実行されます

オリジナルの記事を6件公開 ・いい ね0 訪問数319

おすすめ

転載: blog.csdn.net/qq_43452745/article/details/102593648
おすすめ