如何快速安装cuda10.2版本的torch-1.10.0

前言

目前官网不能直接支持安装cuda10.2

安装过程

创建环境

创建了一个python版本为3.8的torch_test的环境
conda create -n torch_test python=3.8

下载对应whl

在https://download.pytorch.org/whl/torch_stable.html找到对应版本的whl，
我安装的是cu102+python3.8版本的。

pip 安装

先切换环境到torch_test

然后通过pip命令安装：

测试

torch.cuda.is_available()测试cuda是否可以用。

安装成功