torch报错AssertionError: Torch not compiled with CUDA enabled解决方法

2023-12-15 15:53:50

参考:
https://blog.csdn.net/m0_46948660/article/details/129205116

编程环境:
windows11, conda

报错处理:

进入环境:
conda activate xxx
安装torch:
conda install torch
引入及测试:
import torch
print("是否可用:", torch.cuda.is_available())        # 查看GPU是否可用
print("GPU数量:", torch.cuda.device_count())        # 查看GPU数量
print("torch方法查看CUDA版本:", torch.version.cuda)  # torch方法查看CUDA版本
print("GPU索引号:", torch.cuda.current_device())    # 查看GPU索引号
print("GPU名称:", torch.cuda.get_device_name(1))    # 根据索引号得到GPU名称

文章来源:https://blog.csdn.net/sinat_26809255/article/details/135016123
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。