PyTorch 设置使用GPU

来源:互联网 发布:java接收微信事件推送 编辑:程序博客网 时间:2024/06/04 01:05
CUDA_VISIBLE_DEVICES=1 python train.py
0 0
原创粉丝点击