pytorch使用cuda报错RuntimeError: CUDA error: unknown error
pytorch使用cuda报错RuntimeError: CUDA error: unknown error
加上torch.cuda.current_device()
例如
import os
import torch
import torch.nn as nn
import argparse
from sklearn import metrics
from torch.utils....
原创
2019-12-17 19:40:13 ·
4267 阅读 ·
1 评论