pytorch指定gpu

參考：https://www.jianshu.com/p/a014016723d8   1. 改變系統變量使得僅目標顯卡可見 export CUDA_VISIBLE_DEVICES=0 #這裏是要使用的GPU編號，正常的話是從0開始 在程序開頭設置os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3' CUDA_VISIBLE_DEVICES=0,1,2,3

>>阅读原文<<