Невозможно получить cuda, доступный в Jupyter Notebook или Spyder для PyTorch

Question

Я установил pytorch cuda с pip и conda

, когда я запускаю эту команду в IDLE:

>> import torch

>> torch.cuda.is_available()

Я получаю «True», но в Spyder или Jupyter Notebook это дает как "False" даже после обновления пакета и conda. Обновление conda для pytorch cuda было с 10.1 до 10.2

Answer 1

Это можно решить, создав другую среду в conda, а затем установив torch

тип:

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

yourenvname - это имя среды

python = xx равно python версия для вашей среды

активируйте среду, используя:

>conda activate yourenvname

, затем установите PyTorch с помощью cuda:

>conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

откройте «spyder» или «jupyter notebook» проверьте, установлен ли он, введите:

> import torch
> torch.cuda.is_available()