windows安装cuda/pytorch

安装cuda

笔者安装指定版本cuda的目的是安装gpu版本的pytorch，故先访问torch官网，查看需要的cuda版本。
访问：

https://pytorch.org/get-started/locally/

可以看到，在windows10下，使用conda安装，需要CUDA11.3
查看已经安装的cuda版本：

nvcc --version

如果命令不存在，或不符合要求，就要下载并安装对应版本的CUDA。
访问：

https://developer.nvidia.com/cuda-11.3.0-download-archive

根据你的平台，选择下载cuda版本。这里笔者下载windows版，等待下载完成，安装即可。

安装pytorch

首先创建虚拟环境：

conda create --name python39_torch1_9 python=3.9

注意创建时不能启用vpn，否则会报host_name的错误。
创建完毕后启用：

conda activate python39_torch1_9

从torch官网就能得到windows的安装命令：

https://pytorch.org/get-started/locally/   # 最新版本
https://pytorch.org/get-started/previous-versions/ # 历史版本

这里，我们选择历史版本：

conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.3 -c pytorch -c conda-forge

等待安装完成即可。