我不能使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路径被添加
1条答案
按热度按时间tvokkenx1#
欢迎来到SO！我怀疑你需要安装一个包含CUDA二进制文件的
pytorch版本，默认情况下pip install torch安装一个pytorch的CPU版本。首先，卸载pytorch：
pip uninstall torch您可以转到以下页面查找适合您的平台的正确安装命令：https://pytorch.org/get-started/locally/