前言
在csdn上找了很多种方法,也尝试了多种方法,最后虽然安装成功的,但是并没有安装到对应的版本,也并未导入到对应的环境中,甚至没搞清楚究竟是哪一种方法让我成功安装gpu版本的pytorch。但好在最后在同学的帮助下,学会了从本地导入安装的方法。以下是此方法的介绍。
前提
1.想在windows系统中安装pytorch的gpu版本,必须要有NVIDIA的显卡支持。
2.在安装pytorch之前,需要有一些准备工作。包括anaconda,cuda,cudnn三者的安装。具体安装步骤详见(Windows 10环境下TensorFlow(gpu版本)配置教程——[图解] [详细版][零基础]_张小胖的博客-优快云博客)
第一步:查询对应的版本
(注:一定要根据自己的cuda与python版本安装对应的torch和torchvision)