along/HPCC2025
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

就用cpu训练网络

Browse Source
This commit is contained in:
weixin_46229132 2025-03-14 09:45:46 +08:00
parent 64935bf92f
commit c1eb9d9528
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
PPO2/PPO.py
View File

@ -7,12 +7,12 @@ from torch.distributions import Categorical
print("============================================================================================")
# set device to cpu or cuda
device = torch.device('cpu')
if(torch.cuda.is_available()):
device = torch.device('cuda:0')
torch.cuda.empty_cache()
print("Device set to : " + str(torch.cuda.get_device_name(device)))
else:
print("Device set to : cpu")
# if(torch.cuda.is_available()):
# device = torch.device('cuda:0')
# torch.cuda.empty_cache()
# print("Device set to : " + str(torch.cuda.get_device_name(device)))
# else:
# print("Device set to : cpu")
print("============================================================================================")