2020年10月16日
PyTorch GPU使用方式
摘要： 首先通过： torch.cuda.is_available() 看你的pytorch是否支持CUDA计算，确认支持后： 1.在终端执行程序时设置使用GPU: CUDA_VISIBLE_DEVICES=1 python train.py Environment Variable Syntax Resu 阅读全文
posted @ 2020-10-16 12:50 OliYoung 阅读(6196) 评论(0) 推荐(0)