从零开始配置深度学习环境:CUDA+Anaconda+Pytorch+TensorFlow
本文适用于电脑有GPU(显卡)的同学,没有的话直接安装cpu版是简单的。CUDA是系统调用GPU所必须的,所以教程从安装CUDA开始。
CUDA安装
CUDA是加速深度学习计算的工具,诞生于NVIDIA公司,是一个显卡的附加驱动。必须使用NVIDIA的显卡才能安装,可以打开任务管理器查看自己的硬件设备。
下载CUDA
简便起见,可以直接通过我提供的下载链接进行下载(V11.6)
通过这个链接可以下载任意CUDA版本:CUDA Toolkit Archive | NVIDIA Developer
下载任意版本cuDNN(需要注册英伟达俱乐部才能直接下载): cuDNN Archive | NVIDIA Developer
推荐使用迅雷进行下载,更详细的下载教程参考:
cuda安装 (windows版)_何为xl的博客-CSDN博客_windows安装cuda](https://blog.csdn.net/weixin_43848614/article/details/117221384)
CUDA的版本选择是跟显卡型号有关还是驱动有关?
一般是驱动版本决定了能用的CUDA版本的上限,比如新版的显卡驱动可以支持老的CUDA runtime。但是老的显卡可能无法更新到最新的显卡驱动
安装CUDA
这边演示CUDA11.6版本的安装。注意:30系列的显卡必须使用CUDA11.0以上的版本,其他显卡既可以使用10也可以使用11版本。
双击打开安装包(这个路径就别动了)
CUDA是加速深度学习计算的工具,诞生于NVIDIA公司,是一个显卡的附加驱动。必须使用NVIDIA的显卡才能安装,可以打开任务管理器查看自己的硬件设备。这边演示CUDA11.6版本的安装。注意:30系列的显卡必须使用CUDA11.0以上的版本,其他显卡既可以使用10也可以使用11版本。
双击打开安装包(默认安装路径就别修改了)
点击同意并继续
选择自定义,点击下一步
这里可以更改安装路径,可以新建2个文件夹,将CUDA Documentation
和CUDA Development
安装在一个文件夹CUDA
,Samples
安装在另一个文件夹sample
。我是安装在D盘
的,这样可以节省C盘空间,不过需要记得之后修改环境变量修改为相应文件目录。
可以使用cmd,输入:
nvcc -V
查看CUDA是否安装成功
用cudnn打补丁
CUDA需要配合cudnn才能正常工作,将cudnn的四个文件(夹),复制到CUDA的安装路径即可,即CUDA文件夹。
将之前下载的压缩包解压到CUDA文件夹里面
添加环境变量
往系统环境变量中的 path 添加如下路径(根据自己的路径进行修改)
C:\Program Files\NVIDIA\CUDA\bin
C:\Program Files\NVIDIA\CUDA\include
C:\Program Files\NVIDIA\CUDA\lib
C:\Program Files\NVIDIA\CUDA\libnvvp
环境变量配置结果应当如下:
检验:
配置完成后,我们可以验证是否配置成功,主要使用CUDA内置的deviceQuery.exe
和 bandwidthTest.exe
:
首先win+R启动cmd,cd到安装目录下的 …\extras\demo_suite
,然后分别执行.\bandwidthTest.exe
和.\deviceQuery.exe
,应该得到下图.
cd C:\Program Files\NVIDIA\CUDA\extras\demo_suite
.\bandwidthTest.exe
.\deviceQuery.exe
CUDA安装到这里就全部完成了.
安装anaconda
官网下载anaconda速度太慢,建议使用镜像网站下载,另外不建议下载最新的anaconda版本,默认的base环境是python 3.10,建议下载早一点的版本(22年),安装没有其他注意事项,按照默认设置安装即可。
清华anaconda镜像下载网站:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=D
Pytorch安装
首先前往Pytorch官网查找适合自己CUDA版本的安装命令。安装命令分为conda命令和pip命令,conda命令不能手动添加镜像,需要更改配置文件,在已经安装好CUDA的基础上可以直接使用pip命令成功率较高(pip命令网络更好)。
Pytorch下载官网:Previous PyTorch Versions | PyTorch
对于我们安装的CUDA11.6,在anaconda提供的命令行工具Anaconda Powershell Prompt
中输入以下命令进行安装
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
成功安装后进行检验:
python
import torch
torch.cuda.is_available()
如果结果为True那么证明GPU版本的pytorch已经安装完成
TensorFlow安装
TensorFlow更新速度较慢,安装命令是通用的,可以尝试在当前环境直接安装,如果报错直接尝试第二种conda安装命令。
pip install --ignore-installed --upgrade tensorflow-gpu
但是通过手动安装的方式可能会遇到许多问题,包括但不限于c++库有问题.
首先,我想说的是,千万不要用pip安装tensorflow-gpu,因为gpu版本需要很多依赖包,包括cuda、cudnn等,而且,就算你单独去下载这些,安装后还是会有很多配置问题导致有误,无法安装成功,所以直接用conda安装(这里留下了一行行泪水)。
因为conda自带的安装源速度特别慢,所以如果你不设置镜像源下载几乎还是安装不了,中途会自动断开,这里我选择清华镜像源,首先打开anaconda prompt,依次输入以下四个指令添加镜像源。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --show channels
成功率最高的安装方式是:这一行代码安装基本不会报错。
conda create -n tensorflow-gpu tensorflow-gpu
安装完成后进行测试:
conda activate tensorflow-gpu
python
import tensorflow as tf
tf.test.is_gpu_available()
参考文章:
关于国内anaconda镜像站点看这一篇就够啦 - 知乎 (zhihu.com)
已解决error: subprocess-exited-with-error_袁袁袁袁满的博客-CSDN博客
Microsoft C++ 生成工具 - Visual Studio
(1条消息) Tensorflow-gpu安装超详细!!!_tensorflow gpu_东海扬尘_BingO的博客-CSDN博客
从零开始配置深度学习环境:CUDA+Anaconda+Pytorch+TensorFlow的更多相关文章
- ubuntu18+gtx1060 +cuda9+cudnn-v7+opencv3.1.0 配置深度学习环境
将笔记本的ubuntu系统更新到18版本后重新配置深度学习环境,在此记载方便日后参考 具体配置为 Ubuntu18.04+gtx1060+opencv-3.1 第1步 安装依赖包 sudo apt-g ...
- 安装 Win10 & Ubuntu 16.04 双系统以及 Ubuntu 配置深度学习环境记录
0. 前言 坑爹的Ubuntu晚上运行还是好好的,第二天中午的时候打开机器发现屏幕分辨率不正常了:2K屏显示800*600左右的分辨率(无法调节),一个图标一拳头大,窗口和网页显示不全.Google查 ...
- 【系统配置】Ubuntu和Windons系统安装配置深度学习环境
Ubuntu系统 1.备份 在服务器上整个装系统之前,需要做好一个工作,也就是相关重要数据的备份,这里主要是将固态中的数据备份到机械硬盘或移动硬盘里,可能在备份的过程中会遇到无法写入的问题,是因为文件 ...
- Windows配置深度学习环境详细教程(一):安装Pycharm和Miniconda、conda环境介绍
序言 对于想要入门Python或者深度学习的初学者而言,配置环境一直是一个令人头疼的问题.相信有许多人经历过安装第三方包失败,安装好了却在使用中报错,安装CUDA.tensorflow.pytorch ...
- [深度学习] ubuntu18.04配置深度学习环境笔记
最近装过很多ubuntu18.04系统的nvidia驱动,cuda10.2,cudnn7.6.5,发现每次都会出现一些小问题.总结了具体步骤,做个记录.主要分为三个步骤:驱动安装,cuda安装,cud ...
- ubuntu16.04下安装配置深度学习环境(Ubuntu 16.04/16.10+ cuda7.5/8+cudnn4/5+caffe)
主要参照以下两篇博文:http://blog.csdn.net/g0m3e/article/details/51420565 http://blog.csdn.net/xuzhongxiong/a ...
- ubuntu16.04下安装配置深度学习环境(一、cuda7.5的安装)
1.下载所需要的软件 cuda7.5下载(点击下载链接),cudnn4.0下载 2.安装NVIDIA驱动. 一般有两种方法:1)一种方法是利用"软件和更新"来安装,依次选择 系统设 ...
- ubuntu16.04上安装深度学习基本框架caffe2 pytorch tensorflow opencv
anaconda3.5.2.0----python3.6: conda install tensorflow-gpu -y --prefix /media/wkr/diskHgst/ubun ...
- deepin深度学习环境配置
deepin是一个精致优美的系统.最近因为工作需要在deepin上配置深度学习环境,话不多说,接下来记录下整个的配置过程. ×××本篇文章适合对深度学习环境配置有一定了解且对deepin系统感兴趣的同 ...
- Ubuntu深度学习环境搭建 tensorflow+pytorch
目前电脑配置:Ubuntu 16.04 + GTX1080显卡 配置深度学习环境,利用清华源安装一个miniconda环境是非常好的选择.尤其是今天发现conda install -c menpo o ...
随机推荐
- java接口自动化需要的技术
1.testNG需要了解的知识 ITestContext这个类可以直接在方法参数里使用,主要作用是可以通过它的context.getSuite()直接获取suite的相关信息.还可以通过它的 cont ...
- Java的由来
Java 发展史 1.1.起源 20 世纪 90 年代,单片式计算机系统诞生,单片式计算机系统不仅廉价,而且功能强大,使用它 可以大幅度提升消费性电子产品的智能化程度. SUN 公司为了抢占市场先机, ...
- jetson TX2 + opencv3.4 + python3 + 双目 +人脸检测
淘宝看到一款很便宜的双目,150元,就买了.想着用它学习一下opencv,好换个工作.当然,也想着能否用它做一些好玩的,比如三维重建之类高大上的东西.先用便宜的入个门,等以后眼界高了再看是不是买那些更 ...
- windows电脑设置【定时任务】开机、关机、启动程序
windows电脑设置[定时任务]开机.关机.启动程序 shutdown -s -t **(延时秒数):关机 shutdown -r -t **(延时秒数):重启 gina -s -t 36 (开机) ...
- vue-devtools 打开 vscode 可能会报错
据说 99% 的人不知道 vue-devtools 还能直接打开对应组件文件?本文原理揭秘 mac 电脑在 VSCode command + shift + p,Windows 则是 ctrl + s ...
- java: javacTask: 源发行版 8 需要目标发行版 1.8
idea同一工作空间中不同工程使用不同的jkd版本.在本地idea同时使用jdk1.7和jdk1.8,不同的java工程使用不同的jdk版本,但是在java代码编译时报错,其报错信息为:[java: ...
- unity 开发小技巧之C#篇 ----持续更新
1.ToString()方法是可以重写的 2.Start()方法可以主动调用 3.接口可以直接使用 jiekou?.Invoke() 来调用所有继承该接口的方法 4.Json 列表和服务器需要的数据不 ...
- 文本的格式化标签(粗体,斜体)和 <div>和<span>标签(都是双标签)
上一个笔记有提到各种型号的标题,为了保证文章的美观,又会有除了标题之外的东西,比如粗体,斜体,下划线,删除线和各种分隔 1加粗,<strong><strong/>或者<b ...
- objectarx调用python注意点
1. 用conda先装好虚拟环境,为了兼容cad2010,最高只能用python3.5(已经不维护了), 如果cad高版本建议python3.6+ 2. 能用pip安装的就用pip安装, 不能用pip ...
- Cryptanalyzing and Improving a Novel Color Image Encryption Algorithm Using RT-Enhanced Chaotic Tent Maps
Cryptanalyzing and Improving a Novel ColorImage Encryption Algorithm Using RT-EnhancedChaotic Tent M ...