跑cuda 程序遇到下面错误:
RuntimeError: CUDA out of memory. Tried to allocate 588.00 MiB (GPU 011.00 GiB total capacity8.97 GiB already allocated190.44 MiB free9.00 GiB reserved in total by PyTorch)
运行程序之前,使用nvidia-smi 查看显存有没有被占用,如果有被占用5M以上,可能是显存没有被释放。通过如下命肢镇令来释放显存。
linux上使弯禅用命埋饥尘令行,云主机可以重启电脑
fuser -k /dev/nvidia* 或者 kill $(lsof -t /dev/nvidia*)
2. 减小batch size。
3. 更换更大显存的云主机。
在C/C++Compiler选项的Preprocessor选项卡下:在Additionalincludedirectories下输入$TOOLKIT_DIR$\森腊INC\Luminary就可以了,前提是你必须已经正确安装驱动库文件.库文件的安装方法如下:位置在:你自己闹春尘看液禅吧,绝对可以解决.唉,是自己粗心大意,忘了给main函数入口了,在主机端代码告袭中加上函数声明和主耐渣函数就行了://函数声明
void runTest(int argc, char** argv)
//主函数
int main(int argc, char** argv)
{
runTest(argc,argv)
CUT_EXIT(argc,argv) //退昌友悄出CUDA
}
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)