PyTorch 指定 GPU 进行训练

1 直接在终端中设定：

CUDA_VISIBLE_DEVICES=1 python main.py

复制代码

2 python 代码中设定：

import osos.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

复制代码

学习更多编程知识，请关注我的公众号：

代码的路

凑字数专用凑字数专用凑字数专用凑字数专用凑字数专用凑字数专用凑字数专用凑字数专用凑字数专用

发布于: 刚刚阅读数: 3

关注

公众号：代码的路 2023-01-10 加入

Java、Python、C++、图像处理、深度学习相关知识分享

发布

暂无评论

评论