【pytorch】解决pytorch:Torch not compiled with CUDA enabled

简介: 【pytorch】解决pytorch:Torch not compiled with CUDA enabled

一、问题背景


启用pytorch的cuda支持时,发现程序运行不了,该原因是由于,pytorch默认支持的是cpu,安装支持gpu的版本即可。

报错如下:

Torch not compiled with CUDA enabled

image.png



二、解决方案


  1. 卸载pytorch
pip uninstall torch torchvision torchaudio


image.png



2.安装支持gpu的pytorch(2023年9月15日时,pytorch版本是2.0.1,官方推荐下载的cuda117和cuda118,具体的安装命令,需要结合pytorch官网的实时文档)

 pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 

image.png


三、搞定


再次调试text2vec,配置cuda模式,已经能正常运转了。

image.png


【备注】:text2vec 开启cuda的方法如下:

from text2vec import SentenceModel
text2vec.similarity.device='cuda'
目录
相关文章
|
5月前
|
机器学习/深度学习 并行计算 PyTorch
百度搜索:蓝易云【Pytorch和CUDA版本对应关系】
请注意,上述版本对应关系只是示例,并非详尽无遗。实际上,PyTorch的每个版本通常会支持多个CUDA版本,而具体支持的CUDA版本也可能因操作系统、硬件配置等因素而有所不同。因此,在使用PyTorch时,建议参考PyTorch官方文档或社区支持的信息,以获取最准确和最新的PyTorch与CUDA版本对应关系。
116 2
|
15天前
|
机器学习/深度学习 并行计算 算法框架/工具
Anaconda+Cuda+Cudnn+Pytorch(GPU版)+Pycharm+Win11深度学习环境配置
Anaconda+Cuda+Cudnn+Pytorch(GPU版)+Pycharm+Win11深度学习环境配置
|
6月前
|
并行计算 Ubuntu PyTorch
Ubuntu 18.04 + CUDA 11.3.0 + CUDNN 8.2.1 + Anaconda + Pytorch 1.10(下)
Ubuntu 18.04 + CUDA 11.3.0 + CUDNN 8.2.1 + Anaconda + Pytorch 1.10(上)
151 0
|
6月前
|
并行计算 Ubuntu PyTorch
Ubuntu 18.04 + CUDA 11.3.0 + CUDNN 8.2.1 + Anaconda + Pytorch 1.10(上)
Ubuntu 18.04 + CUDA 11.3.0 + CUDNN 8.2.1 + Anaconda + Pytorch 1.10
162 0
|
6月前
|
并行计算 Ubuntu PyTorch
Xshell远程连接配置 Ubuntu 18.04.6 + Anaconda + CUDA + Cudnn + Pytorch(GPU+CPU)(上)
Xshell远程连接配置 Ubuntu 18.04.6 + Anaconda + CUDA + Cudnn + Pytorch(GPU+CPU)
133 0
|
7月前
|
并行计算 PyTorch Linux
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统下GPU版PyTorch(CUDA 12.1)快速安装
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统下GPU版PyTorch(CUDA 12.1)快速安装
2631 0
|
8月前
|
机器学习/深度学习 并行计算 PyTorch
深度学习|如何确定 CUDA+PyTorch 版本
深度学习|如何确定 CUDA+PyTorch 版本
450 0
|
9月前
|
并行计算 PyTorch 算法框架/工具
Pytorch安装和检验cuda和cudnn是否可用
Pytorch安装和检验cuda和cudnn是否可用
556 0
|
9月前
|
机器学习/深度学习 并行计算 PyTorch
CUDA和显卡驱动以及pytorch版本的对应关系
CUDA和显卡驱动以及pytorch版本的对应关系
1680 0
|
11月前
|
并行计算 PyTorch 算法框架/工具
【PyTorch】cuda()与to(device)的区别
【PyTorch】cuda()与to(device)的区别
146 0