PyTorch指定GPU进行训练

时间：2023-03-26 11:49:30 Python

原文链接1直接在终端设置：CUDA_VISIBLE_DEVICES=1pythonmain.py2在python代码中设置：importosos.environ["CUDA_VISIBLE_DEVICES"]="0,1"学习更多编程知识，请关注我的公众号：代码之道

PyTorch指定GPU进行训练相关文章