2021年5月10日
指定用第几块GPU、CUDA error: invalid device ordinal、
摘要： 在指定用第几块GPU的时候，在代码开头输入了 import os os.environ["CUDA_VISIBLE_DEVICES"] = "2" 表示用第二块GPU，并且也只能看见第二块，然后输入： print('# GPUs = %d' % (torch.cuda.device_count()) 阅读全文
posted @ 2021-05-10 18:46 小千北同学超爱写代码 阅读(3251) 评论(0) 推荐(0) 编辑
常用命令行（自用）
摘要： 进入容器： docker exec -it 80a3cca3b305 /bin/bash 查看gpu使用情况： nvidia-smi 查看torch和gpu配好没有： import torch if torch.cuda.is_available(): print("gpu cuda is avai 阅读全文
posted @ 2021-05-10 14:45 小千北同学超爱写代码 阅读(64) 评论(0) 推荐(0) 编辑