在pytorch中运行py代码出现如下错误，求大神帮助 RuntimeError: CUDA error: unknown error

Posted 2023-05-01

import os import csv TRAIN_IMG_DIR = '/home/yourname/Documents/tensorflow/images/500pics2/train' TRAIN_CSV_DIR = '/home/yourname/Documents/tensorflow/images/500pics2/train_labels.csv' TEST_IMG_DIR = '/home/yourname/Documents/tensorflow/images/500pics2/test' TEST_CSV_DIR = '/home/yourname/Documents/tensorflow/images/500pics2/test_labels.csv' def mkcsv(img_dir, csv_dir): list = [] list.append(['File Name','Label']) for file_name in os.listdir(img_dir): if file_name[0] == '3': #bus item = [file_name, 0] elif file_name[0] == '4': #dinosaur item = [file_name, 1] elif file_name[0] == '5': #elephant item = [file_name, 2] elif file_name[0] == '6': #flower item = [file_name, 3] else: item = [file_name, 4] #horse list.append(item) print(list) f = open(csv_dir, 'w', newline='') writer = csv.writer(f) writer.writerows(list) mkcsv(TRAIN_IMG_DIR, TRAIN_CSV_DIR) mkcsv(TEST_IMG_DIR, TEST_CSV_DIR) 参考技术A 哆啦A梦笋子次数太多东西都是我的意思

ad20安装过程中老出现错误 这问题怎么解决呀？求大神指点

安装过程中老出现错误怎么解决？我觉得你应该问一个专业人士，他会告诉你。 参考技术A 这个出现错误的过程，实际上它应该是重启完成的。 参考技术B 这就是软件与系统冲突造成的，看看是否缺插件。