无法使用pyTorch创建Cuda

iszxjhcz  于 2023-05-29  发布在  其他
关注(0)|答案(1)|浏览(119)

我不能使cuda运行时与 Torch 。
我用这个:

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')  
                model = model.to(device)

我安装了:

torch 2.0.1
VS 2022 17.4.0
Cuda 11.8.0
win 11

并将其用于:

NVIDIA GeForce GTX 1660 Ti

我还错过了什么为什么不跑?
我安装了其他的vs和其他的cuda。。但现在我有目前那些和cuda路径被添加

tvokkenx

tvokkenx1#

欢迎来到SO!我怀疑你需要安装一个包含CUDA二进制文件的pytorch版本,默认情况下pip install torch安装一个pytorch的CPU版本。
首先,卸载pytorch:pip uninstall torch
您可以转到以下页面查找适合您的平台的正确安装命令:https://pytorch.org/get-started/locally/

相关问题