当前位置：网站首页>pytorch, 指定显卡问题

pytorch, 指定显卡问题

2022-07-22 11:05:00 【迷路在代码中】

在服务器上跑PyTorch的训练代码，原来一直都没有问题。但是重新安装cuda之后，无法指定在哪几块GPU上运行了，只能按照顺序从第0块开始使用。
 

解决方法：在 os 之后紧随指定显卡。（因为当导入 torch 库后，其已经调用显卡。）

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "4,5,6,7"
 
import torch

 

 

Game Over!

原网站

版权声明
本文为[迷路在代码中]所创，转载请带上原文链接，感谢
https://blog.csdn.net/chenran187906/article/details/109230633

边栏推荐

猜你喜欢

随机推荐