Pytorch错误：Torch not compiled with CUDA enabled_type torch.cuda.floattensor not available. torch n

作者：凡人多烦事01 | 2024-02-09 12:04:27

踩

type torch.cuda.floattensor not available. torch not compiled with cuda enab

一、问题

今天跑一个项目代码遇到了一个问题，大致是pytorch和CUDA版本不匹配，然后我就降低pytorch版本，结果1.7中网络不包含我所需的函数，我直接pip install pytorch==1.8，这样产生了错误：Torch not compiled with CUDA enabled。

二、解决方法

nvcc -V
1

检查一下当前CUDA版本

根据自己的CUDA版本下载安装whl文件

参考链接：https://download.pytorch.org/whl/torch_stable.html

分别

pip install torch-1.8.1+cu101-cp38-cp38-linux_x86_64.whl
pip install torchcsprng-0.2.1+cu101-cp38-cp38-linux_x86_64.whl
pip install torchvision-0.9.1+cu101-cp38-cp38-linux_x86_64.whl
1
2
3

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/凡人多烦事01/article/detail/71815?site