博客
关于我
Windows10 + cuda10.0 + pytorch1.2.0 + torchvision0.4.0环境配置
阅读量:595 次
发布时间:2019-03-12

本文共 1608 字,大约阅读时间需要 5 分钟。

Windows10 + cuda10.0 + pytorch1.2.0 + torchvision0.4.0环境配置

先附上我使用的文件的百度链接,后续也有下载路径,可以根据自己的需要下载

链接:https://pan.baidu.com/s/1vy7NwfL25_uFvFNpWj19Ig
提取码:fa68
要安装pytorch,首先要配置好python环境,个人建议使用anaconda安装,首先到官网上下载anaconda。
在这里插入图片描述

大概长这样,然后双击运行,一路的next和agree就可以了,最后一步请看图,

在这里插入图片描述

一定要勾上这个勾,虽然他官方不建议,但是其实这个的目的只是为了自动配置环境变量,要是不勾的话就自己配置就可以了,否则是没法用的,所以还是老老实实勾上吧。

安装完之后打开终端,就电脑左下角输入cmd打开就可以,然后在终端中输入python
在这里插入图片描述

如图输入python后会看的安装的版本时间等信息,那么就恭喜你安装完了,然后就可以按照pytorch了。

简单说一下pytorch的环境配置,这里以pytorch1.2.0为例,可以按照这个方法安装任何版本。
首先先要下载cuda和cudnn,先看cuda,附上网址:
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
在这里插入图片描述

如图可以选择自己需要的cuda版本,我选的是cuda10.0,然后再下载cudnn,附网址:

https://developer.nvidia.com/cudnn
在这里插入图片描述

如图点击download,进去之后需要注册登录一下,不过可以微信或者qq登录,还挺人性化,登录之后可以看见这些

在这里插入图片描述

如果需要更多的版本,可以点击Archived cuDNN Releases,然后就能看见这样一翻景象,

在这里插入图片描述

选择自己想要的,后面写的很清楚是,一定要选适合自己的cuda版本的,我选的是7.4.1,然后就可以进行安装了。

双击cuda的安装包,
在这里插入图片描述

最好是不要更改这个地址,直接点击ok

在这里插入图片描述

然后就同意并继续,这里选择自定义,然后一直下一步安装即可,安装好cuda后,将cudnn的压缩包解压,找到这个路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0,可以看到很多的文件夹,找到cudnn中同名的三个文件夹

在这里插入图片描述

将这三个文件夹中的文件分别复制到cuda中的同名文件夹中,cuda的安装就这样愉快的结束了。然后打开终端,输入nvcc -V,即可查看安装的版本。

在这里插入图片描述

折腾了半天,终于可以按照pytorch了,这里给两种方法,第一种是简单的,直接在终端中输入命令:

pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

如果这样的方式安装特别慢,那还有另一种办法,就是直接安装whl文件,那么就需要先下载一下这两个文件了。

https://download.pytorch.org/whl/torch_stable.html
进入这个网址里面可以下载torch和torchvision
在这里插入图片描述
在这里插入图片描述

要下载cu开头的文件,不要下载cpu开头的,而且cu的版本要和自己安装的cuda版本一致。下载好之后,在下载的位置直接打开终端,然后输入

pip install torch-1.2.0-cp36-cp36m-win_amd64.whl
pip install torchvision-0.4.0-cp36-cp36m-win_amd64.whl
全部安装完成之后,别忘了,重启一下电脑。

转载地址:http://qlhtz.baihongyu.com/

你可能感兴趣的文章
Nacos基础版 从入门到精通
查看>>
Nacos如何实现Raft算法与Raft协议原理详解
查看>>
Nacos安装教程(非常详细)从零基础入门到精通,看完这一篇就够了
查看>>
Nacos实战攻略:从入门到精通,全面掌握服务治理与配置管理!(上)
查看>>
Nacos实战攻略:从入门到精通,全面掌握服务治理与配置管理!(下)
查看>>
Nacos心跳机制实现快速上下线
查看>>
nacos报错com.alibaba.nacos.shaded.io.grpc.StatusRuntimeException: UNAVAILABLE: io exception
查看>>
nacos服务提供和发现及客户端负载均衡配置
查看>>
Nacos服务注册与发现demo
查看>>
Nacos服务注册与发现的2种实现方法!
查看>>
nacos服务注册和发现原理简单实现案例
查看>>
Nacos服务注册总流程(源码分析)
查看>>
nacos服务注册流程
查看>>
Nacos服务部署安装
查看>>
nacos本地可以,上服务器报错
查看>>
Nacos注册Dubbo(2.7.x)以及namespace配置
查看>>
Nacos注册中心有几种调用方式?
查看>>
nacos注册失败,Feign调用失败,feign无法注入成我们的bean对象
查看>>
nacos源码 nacos注册中心1.4.x 源码 nacos源码如何下载 nacos 客户端源码下载地址 nacos discovery下载地址(一)
查看>>
nacos源码 nacos注册中心1.4.x 源码 spring cloud alibaba 的discovery做了什么 nacos客户端是如何启动的(二)
查看>>