Win10+Anaconda环境下本地安装Pytorch

Win10+Anaconda环境下本地安装Pytorch

参考文章：Win10+Anaconda环境下安装Pytorch
使用anaconda安装pytorch
Windows10+Anaconda+PyTorch(cpu版本)环境搭建

问题描述：

win10+CUDN10.1+Anaconda+Pytroch
电脑配置：Win10、cudn10.1、GPU、已安装好anaconda
需解决问题：在anaconda下安装Pytrpch
目前安装的方式有两种：①在线②本地

原因分析：

本地安装方法出现的问题：
参考 使用anaconda安装pytorch，选择本地安装，出现问题：
①下载时，经常出现网络超时问题
②下载完成后，只有.tar压缩文件，没有出现.whl文件

借鉴其他方法，仍出现问题。
考虑到电脑配置不同，出现对应版本不同的情况，不能根据网上已有CPU安装的步骤。
故总结出该配置下Pytroch的安装方法。

解决方案：

根据官网提示在anaconda prompt进行下载：

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

安装完成！！

验证方法：检验是否安装成功pytorch

import torch
a = torch.cuda.is_available()
print(a)
ngpu= 1
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

经过验证，安装成功。