PyTorch指定GPU进行训练
【PyTorch指定GPU进行训练】原文链接
1直接终端中设定:
CUDA_VISIBLE_DEVICES=1 python main.py
2 python代码中设定:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
学习更多编程知识,请关注我的公众号:
代码的路
文章图片
推荐阅读
- DeepLearning学习|(pytorch-深度学习)使用pytorch框架nn.RNN实现循环神经网络
- PyTorch|语义分割基础知识
- pytorch|【pytorch笔记】(五)自定义损失函数、学习率衰减、模型微调
- 解决使用|解决使用 Eruda 绑定 dom 未在指定位置显示问题
- Python|Linux服务器配置Python+PyTorch+CUDA深度学习环境
- 深度学习自学笔记|Pytorch常见报错、踩坑点总结(持续更新)
- python|PyTorch入门,帮你构建自己的神经网络模型
- MindSpore易点通·精讲系列--模型训练之GPU分布式并行训练
- 深度学习实战项目|深度学习实战6-卷积神经网络(Pytorch)+聚类分析实现空气质量与天气预测
- 深度学习|PyTorch 具体显存占用分析