Bug: CUDA out of memory. Tried to allocate ...问题

今天跑代码出现以下bug:

CUDA out of memory. Tried to allocate 60.00 MiB (GPU 0; 10.76 GiB total capacity; 9.03 GiB already allocated; 55.94 MiB free; 316.78 MiB cached)

看到实际内存还有20G,排除内存不够的问题;

网上查到是pytorch与cuda版本不匹配原因;

输入命令:watch -n 1 nvidia-smi

发现自己的cuda 版本是10.2。

但是运行 

cat /usr/local/cuda/version.txt

显示如下:

CUDA Version 10.0.130
CUDA Patch Version 10.0.130.1

以第二个为准,(实际安装第一个官网没有安装教程)

 安装命令如下:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

 在测试。。。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章