最近人工智能等多门课需要复现论文,近两年的论文很多都是基于Pytorch环境做的实验,所以,这里总结一下Pytorch的安装教程,做好最快、最简单、最好地完成安装。
本机环境 |
---|
Win10+1050Ti+Python3.7 |
1、查看本机的CUDA版本
cmd命令行输入nvidia-smi
,在第一行最右边可以看到CUDA的版本号,我的版本是11.1
2、安装Pytroch
然后选择Get Started,就是如下界面
2、这里进行Pytorch版本的选择,首先我选择的是Stable稳定版,然后OS是Windows系统,Package包就使用Conda,Language肯定选Python,最后的Compute Platform就根据大家的需求来定了。大家如果想在自己电脑(具有NVIDIA显卡)上用GPU进行深度学习模型的推理,就选CUDA;如果只想调用CPU运行算法模型的推理,就选CPU。
一句话总结:包含CUDA的Pytorch可以在程序中设置调用GPU或CPU运行,不含CUDA的Pytorch只能调用CPU运行程序。
选CPU的同学可以直接看第3步了。我们在第一步已经看过自己的NAVIDA的CUDA版本了,这里我们一定要选择比自己版本低的CUDA。比如,像我的版本是11.1,那么就只能选择10.2,因为11.3对我来说有点高了。
当前官网最新版是CUDA11.3,如果你的CUDA版本比官网推荐的都低,可以点击下面的Previous versions of PyTorch,查看以前的版本。
3、安装下anaconda深度学习神器,没有安装的同学,请按照这篇教程安装好anaconda,并创建一个python环境:anaconda安装配置教程,python的版本根据你的需求而定,这里我根据复现算法的需求环境,安装的是python3.7。
4、打开anaconda命令行,先激活需要安装Pytorch的python环境(这里我将python环境命名为pytorch,环境名称随意),复制第二步最下面那段命令行语句,然后回车执行:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
5、可以看到将要安装的包里面有pytorch和cudatoolkit,我们输入y确认,然后回车
6、下面就等待这些package安装成功,可以看到Pytorch还是比较大的,所以耐心等待一下。如果下载过程中卡住的同学,直接到torch镜像源下载安装包进行安装,效果相同。
7、安装完成后,会提示信息done。
8、输入pip list,查看我们的包是否被安装,这里可以看到torch相关的包都安装了。
9、我们输入python
进入下Python环境,然后输入import torch
,如果没有报错说明可以导入成功。
10、输入torch.cuda.is_available()
查看torch是否可以使用显卡,True就代表可以!
安装好Pytorch环境后,我们就可以来部署运行些深度学习算法了,比如经典的Yolov5,Mask-RCNN、Swin Transformer等,快去体验吧~
CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。开发人员可以使用C语言来为CUDA™架构编写程序,所编写出的程序可以在支持CUDA™的处理器上以超高性能运行。
这里给大家分享一份Python全套学习资料,包括学习路线、软件、源码、视频、面试题等等,都是我自己学习时整理的,希望可以对正在学习或者想要学习Python的朋友有帮助!
如果大家对Python感兴趣,这套python学习资料一定对你有用
对于0基础小白入门:
如果你是零基础小白,想快速入门Python是可以考虑的。
一方面是学习时间相对较短,学习内容更全面更集中。
二方面是可以根据这些资料规划好学习计划和方向。
包括:Python激活码+安装包、Python web开发,Python爬虫,Python数据分析,人工智能、机器学习、Python量化交易等习教程。带你从零基础系统性的学好Python!
零基础Python学习资源介绍
① Python所有方向的学习路线图,清楚各个方向要学什么东西
② 600多节Python课程视频,涵盖必备基础、爬虫和数据分析
③ 100多个Python实战案例,含50个超大型项目详解,学习不再是只会理论
④ 20款主流手游迫解 爬虫手游逆行迫解教程包
⑤ 爬虫与反爬虫攻防教程包,含15个大型网站迫解
⑥ 爬虫APP逆向实战教程包,含45项绝密技术详解
⑦ 超300本Python电子好书,从入门到高阶应有尽有
⑧ 华为出品独家Python漫画教程,手机也能学习
⑨ 历年互联网企业Python面试真题,复习时非常方便
👉Python学习路线汇总👈
Python所有方向的技术点做的整理,形成各个领域的知识点汇总,它的用处就在于,你可以按照上面的知识点去找对应的学习资源,保证自己学得较为全面。(全套教程文末领取哈)
👉Python必备开发工具👈
温馨提示:篇幅有限,已打包文件夹,获取方式在:文末
👉Python学习视频600合集👈
观看零基础学习视频,看视频学习是最快捷也是最有效果的方式,跟着视频中老师的思路,从基础到深入,还是很容易入门的。
👉实战案例👈
光学理论是没用的,要学会跟着一起敲,要动手实操,才能将自己的所学运用到实际当中去,这时候可以搞点实战案例来学习。
👉100道Python练习题👈
检查学习结果。
👉面试刷题👈
资料领取
这份完整版的Python全套学习资料已经上传网盘,朋友们如果需要可以点击下方微信卡片免费领取 ↓↓↓【保证100%免费】