问题等同于“torch.cuda.is_available() = false”
之前怎么安装的pytorch已经不记得了,最近需要加速代码,想使用GPU. 但是torch.cuda.is_available() 总是显示 False. 为此决定从新安装PyTorch, CUDA 等。
前提:
- 了解自己pc上nvidia驱动所支持的最高CUDA version.
- 方法: ctrl+R -> cmd -> nvidia-smi
操作:
1. 在previous版本中找到合适的pytorch版本,根据pc OS和pip or conda复制安装命令。
例如v1.13.1.
- 这样做的好处在于,我们不需要操心pytorch, cuda的版本适配问题。
2. 这样安装之后我出现了以下问题
解决方案:
conda install cudatoolkit
- 升级一下cudatoolkit. 注意这里版本需要低于文章“前提”中提到的cuda version.
检查:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
# results are
1.13.1
True