- 博客(5)
- 收藏
- 关注
原创 【os.environ[“CUDA_VISIBLE_DEVICES“]使用测试】
os.environ[“CUDA_VISIBLE_DEVICES”]前提:测试设备共有4块显卡测试:import os# os.environ["CUDA_VISIBLE_DEVICES"]设置程序可见的显卡号# 设置可见显卡为1,2,3号, 0号显卡不可见os.environ["CUDA_VISIBLE_DEVICES"] = "1, 2, 3" import torchx = torch.randn(5)# 测试y = x.to("cuda:0")y = x.to("cuda:
2022-01-18 15:08:46
4526
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人