关于SNN实现手写数字识别的代码的问题

[Spice-monkey]

您好，我使用您的示例代码时，出现了一些错误，主要是关于这个torch.cuda中amp的问题,"GradScaler" is not exported from module "torch.cuda.amp"，还有是在252行出现的Type "list[Unknown]" cannot be assigned to type "Tensor | Module"，小白一枚还请赐教

[fangwei123456]

安装的pytorch支持cuda吗

[Spice-monkey]

我去确认一下

[Spice-monkey]

显示是True的

[Spice-monkey]

我更新了貌似还是不行，请您看到有空回复下，打扰您了十分感谢！

[fangwei123456]

提供最少的能够复现错误的代码

[fangwei123456]

提供最少的能够复现错误的代码，不要以图片形式。提供你运行的输出错误

[Spice-monkey]

好的 稍等

[Spice-monkey]

Traceback (most recent call last):
File "e:\迅雷下载\lif_fc_mnist (2).py", line 278, in
main()
File "e:\迅雷下载\lif_fc_mnist (2).py", line 77, in main
train_dataset = torchvision.datasets.MNIST(
File "E:\CS\Anaconda3\envs\SNN\lib\site-packages\torchvision\datasets\mnist.py", line 94, in init
if self._check_legacy_exist():
File "E:\CS\Anaconda3\envs\SNN\lib\site-packages\torchvision\datasets\mnist.py", line 107, in _check_legacy_exist
processed_folder_exists = os.path.exists(self.processed_folder)
File "E:\CS\Anaconda3\envs\SNN\lib\site-packages\torchvision\datasets\mnist.py", line 161, in processed_folder
return os.path.join(self.root, self.class.name, "processed")
File "E:\CS\Anaconda3\envs\SNN\lib\ntpath.py", line 78, in join
path = os.fspath(path)
TypeError: expected str, bytes or os.PathLike object, not NoneType

[fangwei123456]

不是SJ框架本身的问题，是你数据集路径设置错误

[fangwei123456]

建议先熟悉pytorch的使用后，再使用SJ框架

[fangwei123456]

另外，建议先问问chatgpt看看他怎么说...

[Spice-monkey]

您好 我发现我的torch版本是2.0.1 是这个原因导致冲突的吗

[fangwei123456]

应该不是。你需要提供一下可以复现错误的代码和报错信息

[Spice-monkey]

貌似是没有装cuda，我忘记了，以前以为装过了，尴尬