写点什么

PyTorch 指定 GPU 进行训练

作者:代码的路
  • 2023-01-15
    江苏
  • 本文字数:138 字

    阅读完需:约 1 分钟

原文链接


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++、图像处理、深度学习相关知识分享

评论

发布
暂无评论
PyTorch指定GPU进行训练_Python_代码的路_InfoQ写作社区