Pytouch GPUは十分に現存していますが、out of memoryの解決方法を示しています。


今日はpytouchコードをテストする時に表示が不足していますが、このネットフレームは簡単なのに、CPUを使って走っても大丈夫です。GPUはずっとout of memoryを提示しています。
ネットでいろいろな方法を探しましたが、結局はpytochバージョンの問題だと思います。元々は私のpytochバージョンは0.4.1だったので、このバージョンをアンインストールして、pytoch 1.1.0をインストールしたら、プログラムは不思議な運行ができます。具体的な原因はまだ分かりませんが、ここはまずマークします。具体的な過程は以下の通りです。
古いバージョンのpytochをアンマウント:

conda uninstall pytorch
pytouch 1.1.0をインストールして、公式サイトの方法によって、私のCUDAバージョンは9.0です。

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
大きな仕事が完成しました
以上のPytouch GPUは十分に現存していますが、out of memoryの解決方法は小編集で皆さんに共有された内容です。参考にしていただければと思います。どうぞよろしくお願いします。