欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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')