今天跑代码出现以下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
在测试。。。