Pytorch相关用法
目录将模型运行在指定GPU上,方便并行化
将模型运行在指定GPU上,方便并行化
这里使用的不是通过设置CUDA_VISIBLE_DEVICES环境变量实现的:
os.environ["CUDA_VISIBLE_DEVICES"]="1,2,3,4"
而是通过指定model和tensor的cuda属性完成,实现过程如下:
import torch
gpuid = "3"
#必须要加上下面这句,否...
原创
2019-12-18 18:01:08 ·
214 阅读 ·
0 评论