下载cuda,大概2G,一路确认安装
nvcc -V #查看安装是否成功
nvidia-smi #查看版本
pytorch官网查看复制pytorch安装命令
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
测试安装是否成功
import torch
torch.cuda.is_available() #输出True则安装成功
下载cuda,大概2G,一路确认安装
nvcc -V #查看安装是否成功
nvidia-smi #查看版本
pytorch官网查看复制pytorch安装命令
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
测试安装是否成功
import torch
torch.cuda.is_available() #输出True则安装成功