机器上有八张显卡,占用了两三张。 选择空闲的GPU后仍报告错误。
使用以下代码检查可用的GPU时,n_gpu为1,显示问号
n_gpu=torch.cuda.device_count ) )解析:在主文件的开头加上此代码,再次查看可用的GPU时为8
import osos.environ [ ' cuda _ visible _ devices ' ]=' 0,1,2,3,4,5,6,7 '
机器上有八张显卡,占用了两三张。 选择空闲的GPU后仍报告错误。
使用以下代码检查可用的GPU时,n_gpu为1,显示问号
n_gpu=torch.cuda.device_count ) )解析:在主文件的开头加上此代码,再次查看可用的GPU时为8
import osos.environ [ ' cuda _ visible _ devices ' ]=' 0,1,2,3,4,5,6,7 '
版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。