python - 无法在 Jupyter Notebook 或 Spyder for PyTorch 中获得 cuda

Question

我已经用 pip 和 conda 安装了 pytorch cuda

当我在 IDLE 中运行此命令时:

>> import torch

>> torch.cuda.is_available()

我得到“真实”，
但是在 Spyder 或 Jupyter Notebook 中，即使在更新包和 conda 之后，它也会显示为“False”。
pytorch cuda 的 conda 更新从 10.1 到 10.2

Answer 1

可以通过在conda中创建另一个环境然后安装torch来解决

类型:

>conda create -n yourenvname python=x.x anaconda

yourenvname 是环境名称

python=x.x 是适用于您环境的 python 版本

使用以下方法激活环境:

>conda activate yourenvname

然后用 cuda 安装 PyTorch:

>conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

打开“spyder”或“jupyter notebook”
验证是否已安装，输入:

> import torch
> torch.cuda.is_available()