出现这个问题怎样解决呀,还有这个是我在GitHub中找的模型,他已经有了数据集。那我如果要训练我自己的数据集是不是就直接该那个数据库还有分类就行了呀。谢谢各位。
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
我的CUDA 版本是11.6
torch版本是1.12.1
GPU是3050笔记本电脑版
出现这个问题怎样解决呀,还有这个是我在GitHub中找的模型,他已经有了数据集。那我如果要训练我自己的数据集是不是就直接该那个数据库还有分类就行了呀。谢谢各位。
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
我的CUDA 版本是11.6
torch版本是1.12.1
GPU是3050笔记本电脑版