【cpu与gpu】

最新推荐文章于 2024-10-05 12:00:56 发布

YUXI2021

最新推荐文章于 2024-10-05 12:00:56 发布

阅读量80

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/yuelite/article/details/130561422

版权

当使用PyTorch尝试运行CUDA时，可能会遇到RuntimeError:Attempthasbeenmadetostartanewprocessbeforethepreviousonefinished...。该问题可能由num_workers多线程参数引起，解决方案是移除该参数或者在代码中添加ifname==main:检查，确保只在直接运行脚本时执行CUDA相关代码。

摘要由CSDN通过智能技术生成

import torch
torch.cuda.is_available() #是否配置cuda
device=torch.device("cuda:0" if torch.cuda.is_available else 'cpu')
print(device)#gpu还是cpu
torch.cuda.get_device_name()#显卡名称

GPU
错误：
pytorch使用出现"RuntimeError: An attempt has been made to start a new process before the…" 解决方法
（1）num_workers多线程参数问题，去除即可
（2）在跑epoch之前，加上if name==‘main’:
当.py文件被直接运行时，if name == 'main’之下的代码块将被运行；
当.py文件以模块形式被导入时，if name == 'main’之下的代码块不被运行。