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]='2,3'
プログラムは正常に実行されます