目录
- 1. 问题
- 2. 解决方法
- 2.1 安装torch的CUDA版本
- 2.1.1 查看安装的torch是什么版本
- 2.1.2 下载GPU版本的torch
- 2.2 安装 NVIDIA CUDA
- 2.2.1 下载CUDA Toolkit
- 2.2.2 安装CUDA Toolkit
- 最终效果
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文件长这样：
安装过程如下：
最终效果
祝各位成功！
还有什么问题请留言讨论！
