1. 问题

在Python中，使用Pytorch时，往往会选择使用CUDA模块，但import torch后，cuda却无法使用，比如：

2. 解决方法

2.1 安装torch的CUDA版本

2.1.1 查看安装的torch是什么版本

如果你是默认pip install torch，它会默认给你安装cpu版本，先输入

pip list

查看安装的torch是什么版本，如果你的是这样，那就不行

需要的torch版本长这样：

2.1.2 下载GPU版本的torch

官网下载链接：https://pytorch.org/

2.2 安装 NVIDIA CUDA

2.2.1 下载CUDA Toolkit

进入CUDA官网下载CUDA Toolkit https://developer.nvidia.com/cuda-downloads，选择合适你的配置

2.2.2 安装CUDA Toolkit

下载好的.exe文件长这样：

安装过程如下：

最终效果

祝各位成功！
还有什么问题请留言讨论！