win10 安装pytorch10.2 gpu成功

安装pytorch10.2：

先安装anaconda3。我之前安过所以没下载。需要可以下载安装。

conda的命令行中使用conda create -n pytorch38 python=3.8建立环境。

nvidia-smi检查发现没有安装显卡驱动。

nvidia-smi检查是否安装400以上版本的显卡驱动。

官网上找到nvidia合适的显卡版本进行安装。

nvidia的显卡驱动，我安的是511.79.默认安装就可以。

完毕后安装pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

安装完毕，可能会比较慢。上G的包。我自己安装还好。

安装完毕测试是否可用。

import torch

torch.cuda.is_available()