win10下安装使用pytorch以及cuda9、cudnn7.0

简介: 以后文章更新在 https://oldpan.me 中 pytorch是一个优雅的深度学习库,相比TensorFlow更年轻也更充满潜力,在官方的介绍中pytorch只支持linux和mac,但其实windows也是可以安装以及正常使用的平台: win10(版本1709) CPU:i5-7400 显卡:1060 6G 内容:8G软件: anaconda3 pycharm专业版首先从官网下载相应的cuda和cudnn库,记住下载win10版本的,不要下错。

以后文章更新在 https://oldpan.me

pytorch是一个优雅的深度学习库,相比TensorFlow更年轻也更充满潜力,在官方的介绍中pytorch只支持linux和mac,但其实windows也是可以安装以及正常使用的

平台:
win10(版本1709)
CPU:i5-7400
显卡:1060 6G
内容:8G

软件:
anaconda3
pycharm专业版

首先从官网下载相应的cuda和cudnn库,记住下载win10版本的,不要下错。一些额外的下载信息可以查看我之前的一篇文章(pytorch-0.2成功调用GPU:ubuntu16.04,Nvidia驱动安装以及最新cuda9.0与cudnnV7.0配置):http://blog.csdn.net/iamoldpan/article/details/78459208

windows下安装cuda和cudnn还是非常容易的,下载最新版即可(cuda9.0和与之相对应的cudnn版本)

安装好之后,开始创建虚拟环境,这里建议使用anaconda3,不论是方便性还是拓展性都比只使用pip强。在官网下载anaconda3然后进行安装,创建一个虚拟环境:http://blog.csdn.net/iamoldpan/article/details/77969477(注意在安装的时候需要勾上环境变量的选项,如果没有需要自行在环境变量中加入相应路径(…/anaconda3/Scripts))

创建好虚拟环境后,在pycharm中创建工程,将解释器选择为之前创建的环境(不用pycharm也可以,这里只是做例子,只要有python的简单IDE即可):

这里写图片描述

上图是已经安装完成后的解释器界面,可以看到pytorch和torchvision以及一系列依赖包都已经安装好。

选择解释器后,开始conda环境中只有刚开安装的几个程序包,这是需要安装pytorch需要的依赖包:

注意:一定要在虚拟环境中进行安装,如果在终端中创建了虚拟环境但是无法激活,可以通过conda install -n my-env package指令虚拟环境来进行安装,同样pip安装直接使用虚拟环境中的pip命令进行安装:.../envs/my-envs/Scripts/pip install package,总之要安装到虚拟环境中去。

conda install numpy pyyaml mkl setuptools cmake cffi

因为直接从官网源下载速度很慢,我们可以修改conda的安装源来进行加速:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

然后进行依赖包的安装

安装完依赖包后,开始进行pytorch和torchvision的安装:
首先下载pytorch的pip安装包:https://pan.baidu.com/s/1nvaamrn#list/path=%2F
我安装的是,pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2,即python3.6-cuda8版本的安装包下好后,进入到该文件的目录中,在终端中输入:

pip install pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2

即可进行安装,安装速度一般很快,如果是固态硬盘,5~6s即可安装完毕。
torchvision的安装相对简单很多,是从conda和pip安装即可,只要注意安装到对应的虚拟环境中即可。

大功告成

全部安装完后,如果没有意外:

这里写图片描述

就可以在window中使用pytorch了

目录
相关文章
|
并行计算 安全 PyTorch
win10安装pytorch
win10安装pytorch
467 0
win10安装pytorch
|
6月前
|
PyTorch 算法框架/工具
win10下安装pytorch,torchvision遇到的bug
win10下安装pytorch,torchvision遇到的bug
|
7月前
|
机器学习/深度学习 并行计算 PyTorch
win10安装minconda+pytorch
本文主要说明在win10笔记本上的安装minconda和pytorch,以图文结合的方式记录安装步骤,供参考学习。
172 1
|
机器学习/深度学习 人工智能 PyTorch
Win10+Python3.6下Pytorch安装(基于conda或pip)
Win10+Python3.6下Pytorch安装(基于conda或pip)
728 0
Win10+Python3.6下Pytorch安装(基于conda或pip)
|
机器学习/深度学习 PyTorch 算法框架/工具
win10安装Pytorch【最新版】
win10安装Pytorch【最新版】
171 0
|
并行计算 PyTorch TensorFlow
Win10 安装Anaconda、Pycharm、Tensorflow和Pytorch
Win10 安装Anaconda、Pycharm、Tensorflow和Pytorch
317 0
Win10 安装Anaconda、Pycharm、Tensorflow和Pytorch
|
并行计算 PyTorch 算法框架/工具
Win10 Python3.7 安装pytorch1.5+ mmdetection2.1,搭建mmdetection环境
Win10 Python3.7 安装pytorch1.5+ mmdetection2.1,搭建mmdetection环境
382 0
Win10 Python3.7 安装pytorch1.5+ mmdetection2.1,搭建mmdetection环境
|
2月前
|
算法 PyTorch 算法框架/工具
Pytorch学习笔记(九):Pytorch模型的FLOPs、模型参数量等信息输出(torchstat、thop、ptflops、torchsummary)
本文介绍了如何使用torchstat、thop、ptflops和torchsummary等工具来计算Pytorch模型的FLOPs、模型参数量等信息。
239 2
|
2月前
|
机器学习/深度学习 自然语言处理 监控
利用 PyTorch Lightning 搭建一个文本分类模型
利用 PyTorch Lightning 搭建一个文本分类模型
61 8
利用 PyTorch Lightning 搭建一个文本分类模型
|
2月前
|
机器学习/深度学习 自然语言处理 数据建模
三种Transformer模型中的注意力机制介绍及Pytorch实现:从自注意力到因果自注意力
本文深入探讨了Transformer模型中的三种关键注意力机制:自注意力、交叉注意力和因果自注意力,这些机制是GPT-4、Llama等大型语言模型的核心。文章不仅讲解了理论概念,还通过Python和PyTorch从零开始实现这些机制,帮助读者深入理解其内部工作原理。自注意力机制通过整合上下文信息增强了输入嵌入,多头注意力则通过多个并行的注意力头捕捉不同类型的依赖关系。交叉注意力则允许模型在两个不同输入序列间传递信息,适用于机器翻译和图像描述等任务。因果自注意力确保模型在生成文本时仅考虑先前的上下文,适用于解码器风格的模型。通过本文的详细解析和代码实现,读者可以全面掌握这些机制的应用潜力。
75 3
三种Transformer模型中的注意力机制介绍及Pytorch实现:从自注意力到因果自注意力