如何在Windows10上安装PyTorch

By minirplus on 2021-04-27 in How To

检查Windows 10 WSL版本

1	wsl --list --verbose

如果显示VERSION为2说明WSL版本是WSL2

根据显卡类型下载对应驱动（需要注册开发者账号）

https://developer.nvidia.com/cuda/wsl/download

右键开始图标——》Windows Powershell

1	nvidia-smi

记录下这里的CUDA Version版本

下载对应python版本并安装

https://docs.conda.io/en/latest/miniconda.html

打开Anaconda Prompt (miniconda3)

根据https://pytorch.org/get-started/locally/页面选择对应命令并运行

1	conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

打开Windows Powershell

进入python环境

python

运行测试代码

import torch

x = torch.rand(5, 3)

print(x)

验证是否开启GPU加速

1 2	import torch torch.cuda.is_available()

CUDA Toolkit

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

CUDA等效算力表

https://developer.nvidia.com/cuda-gpus#compute

升级conda

1	conda update -n base -c defaults conda