Anaconda 安装的虚拟环境不能使用CUDA
程序员文章站
2024-03-13 09:18:03
...
在Anaconda的base环境中,torch.cuda.is_available()函数是true,但是转为自己创建的虚拟环境后变为false;具体原因尚未找到,但是可以通过下面操作进行更正:
操作前:
torch.load(model_file)
操作后:
torch.load(model_file, map_location='cpu')
下一篇: php动态读取数据清除最右边距的方法