cuda pytorch 环境变量_Windows10+CUDA 10.1.0+pytorch安装过程
写这篇文章的主要目的是记录和分享一下自己成功安装的过程,如果有小伙伴正在被这个问题困扰的话,也可以看看是否能从这里找到解决的办法。
话不多说,先梳理一下我的大概流程安装CUDA
安装cuDNN
安装pytorch
测试
第一步,安装CUDA
我们都知道安装CUDA是为了利用显卡上的GPU加速运算,所以首先要确定一个事,你的电脑显卡是否支持CUDA,并且是否已经安装了驱动。
先说驱动,打开设备管理器,找到显示适配器,如果能够看到自己的显卡型号的话,就说明驱动已经安装成功了,如果出现黄色感叹号的话,说明驱动异常,需要重新安装。我的电脑是直接安装好了,所以不用再去下载安装,如果你需要安装的话可以自行去官网下载,一些注意事项可以提前搜索一下。
再说如何查看自己的显卡是否支持CUDA,提供一种方法,找到NVIDIA控制面板我的是在右下角状态栏里。然后点帮助,组件,NVCUDA.DLL可以查看显卡支持的CUDA版本。
接下来就是去官网下载对应版本的CUDA,我当时直接去官网,发现直接下载是11版本了,而我需要的是10.1版本,后来通过查询,发现这个地址可以下载任意版本的CUDA。
然后选择按自己的需要选择下载。
下载以后按步骤进行安装就可以了,这里还有一个前提是,你的电脑要提前安装了vs2015版本以上。安装过程中注意有两个路径,第一个是临时解压的路径,第二个才是安装的路径。安装的路径如果不是默认的话,需要记住,因为后面添加环境变量的时候要用到。
上面是第一个路径
选择自定义
这是第二个路径,实际安装的位置,建议不要修改,方便后面添加环境变量。
打开powershell,或者cmd命令窗口,输入nvcc -V,查看是否安装成功。上图表示已经安装成功。
添加环境变量。电脑->属性->高级系统设置->环境变量
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1 CUDA_LIB_PATH = %CUDA_PATH%\lib\x64 CUDA_BIN_PATH = %CUDA_PATH%\bin CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64 CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
将上面的内容新建到系统变量之中
然后再找到系统变量中的path,选择编辑。
添加以下内容:
%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%; C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64; C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\common\lib\x64; C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\bin\win64;
第二步,安装cuDNN
首先去官网下载。第一次需要注册一下,
选择CREATE AN ACCOUNT
然后就是注册一个账号,注册以后再回去下载。
选择与CUDA适合的版本。
再选择与自己系统合适的cuDNN。
下载以后是一个安装包,解压以后会有三个文件夹。将三个文件夹中的内容复制到你安装的CUDA中相应的文件夹中去。比如你解压以后有个lib,那么就复制其中的内容。再找到CUDA的路径,同样找到lib文件夹,将复制的内容到文件夹下。
第三步,安装pytorch
去pytorch官网选择合适的版本,然后复制代码到CMD或者powershell。
安装之前如果以前有安装过torch,pytorch,torchvision统统卸载。
pip uninstall torch
或者用
conda uninstall torch
我当时是之前有安装过,所以直接安装不成功,卸载以后可以。
最后一步,测试
打开powershell,进入pyhton,torch.codu.is_available输出为Ture,则表示安装成功,可以使用了。
本文是自己的安装总结,可能有不是很完善的地方,欢迎指出,交流。
cuda pytorch 环境变量_Windows10+CUDA 10.1.0+pytorch安装过程相关推荐
- 中修改环境变量_Golang入门(1):安装与配置环境变量的意义
摘要 在几年前学习Java的时候,环境的配置就会劝退一部分的初学者.而对于Golang来说,也需要从环境的配置开始学起.这一篇文章将从如何安装Golang开始讲起,随后将会提到Golang中的环境变量 ...
- 基于Windows环境下CPU和GPU版本Tensorflow详细安装过程
基于Windows环境下CPU和GPU版本Tensorflow详细安装过程 一.Tensorflow简介 TensorFlow 是一个开源的.基于 Python 的机器学习框架,它由 Google 开 ...
- oracle 操作系统升级,学习笔记:Oracle升级 linux操作系统10.2.0.1 升级至 10.2.0.3详细过程...
天萃荷净 记录一次用户现场Oracle 10G数据库升级的过程,Linux操作环境 FOR 10.2.0.1 to 10.2.0.3的版本升级过程 数据库版本linux 32位(10.2.0.1升级到 ...
- pytorch 源_Windows10+CUDA 10.1.0+pytorch安装过程
写这篇文章的主要目的是记录和分享一下自己成功安装的过程,如果有小伙伴正在被这个问题困扰的话,也可以看看是否能从这里找到解决的办法. 话不多说,先梳理一下我的大概流程 安装CUDA 安装cuDNN 安装 ...
- CUDA 添加环境变量
按照这种格式添加的CUDA环境变量亲测没毛病 首先打开.bashrc文件: sudo gedit ~/.bashrc 在打开文件后,在末尾处添加如下语句: export PATH="$PAT ...
- mysql配置环境变量(win 10)_mysql配置环境变量(win 10)
1.安装完mysql后就需要配置环境变量 (win 10) 选择"我的电脑",单击右键,选择"属性->高级->环境变量中的系统变量,对 MYSQL_HOME. ...
- mysql配置环境变量(win 10)
1.安装完mysql后就需要配置环境变量 (win 10) 选择"我的电脑",单击右键,选择"属性->高级->环境变量中的系统变量,对 MYSQL_HOME. ...
- java jdk 环境变量配置(window 10 系统)
oracle 官网下载 jdk 安装到一个目录 例如我的安装目录 然后就是打开 环境变量 计算机-右键最下面有个属性- 高级属性 点击环境变量 记得刚开始的时候自己学网上创建JAVA_HOME cla ...
- 带你玩转Jetson Xavier NX系列教程 | Xavier NX 环境变量配置,风扇控制以及Jtop安装
写在最前:转载请注明文章出处.本期文章分为三部分内容,均为刷机以后需要配置的小细节,建议大家刷机以后立刻进行配置,没有配置的后续再配置也没什么大问题. Session_1 : : Xavier NX风 ...
最新文章
- 神经网络入门——12梯度下降代码
- 系统调用和库函数调用的区别
- 源码:我的关于NLP的博客(持续更新中...)
- spark实验遇到的问题
- Offer年薪低于25W全额退款|阿里、腾讯内推快艇《全链路大数据分析工程师》课程招生简章...
- Keras 的预训练权值模型用来进行预测、特征提取和微调(fine-tuning)
- bag文件加载及可视化显示
- 55)PHP,在html嵌套PHP写法
- Linux下的时间戳
- 2017第八届(C/C++)B组蓝桥国赛题
- go get报错 i/o timeout的解决办法
- python 时间格式处理
- 计算机如何一次性删除音乐,win10怎么删除windows音乐文件夹?
- 接口收到CRC错包怎么处理
- Typora+PciGo-Core+SMMS自动上传图片
- 用unity开发贪吃蛇,勾起我童年乐趣的小游戏
- Excel合并单元格如何分组排序?
- “知行合一”到底有什么现实意义?
- 【Kubernetes 企业项目实战】05、基于云原生分布式存储 Ceph 实现 K8s 数据持久化(下)
- 嵌入式软件管培生每日总结-第2天