小千北同学超爱写代码

摘要： 在指定用第几块GPU的时候，在代码开头输入了 import os os.environ["CUDA_VISIBLE_DEVICES"] = "2" 表示用第二块GPU，并且也只能看见第二块，然后输入： print('# GPUs = %d' % (torch.cuda.device_count()) 阅读全文

摘要： 进入容器： docker exec -it 80a3cca3b305 /bin/bash 查看gpu使用情况： nvidia-smi 查看torch和gpu配好没有： import torch if torch.cuda.is_available(): print("gpu cuda is avai 阅读全文