Python Pytorch开发环境搭建(Windows和Ubuntu)
Python Pytorch开发环境搭建(Windows和Ubuntu)
目录
Python Pytorch开发环境搭建(Windows和Ubuntu)
深度学习模型算法比较复杂,如果使用CPU计算,速度会很慢,因而需要使用GPU进行并行计算加速 。深度学习框架,如Pytorch,TensorFlow都支持GPU训练,使用GPU设备需要显卡的支持,比如常见1080显卡,2070显卡等,同时需要安装对应的显卡驱动,以及CUDA和cuDNN库。CUDA 是 NVIDIA 发明的一种并行计算平台和编程模型,它通过利用图形处理器 (GPU) 的处理能力,可大幅提升计算性能。而cuDNN(CUDA Deep Neural Network library)是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。
【尊重原创,转载请注明出处】https://blog.csdn.net/guyuealian/article/details/129163343
1. 安装cuda cudnn
(1)Windows安装方法
- 参考安装教程:Windows10系统下CUDA和cuDNN安装教程
(2)Ubuntu18.04安装方法
- 参考安装教程:ubuntu18.04安装cuda和cudnn
2. 安装Python(推荐使用Anaconda)
(1)Windows安装方法
- 参考安装教程:Anaconda超详细安装教程(Windows环境下)
-
conda使用教程:Windows中Conda 创建和使用虚拟环境
(2)Ubuntu18.04安装方法
3. Pytorch安装
PyTorch是基于Torch的开源Python机器学习库
请选择自己的版本进行安装,比如如果你安装cuda=11.0,那么则安装对应的版本torch
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
4. 安装项目依赖包
项目python依赖包,Windows和Ubuntu安装方法一致;一般项目都自带requirements.txt文件,里面包含了项目开发所需要的python依赖包已经对应的版本号,比如下图第一个依赖包numpy==1.18.5,说明项目使用了numpy库,对应的版本是1.16.3,你可选择使用pip安装对应的版本:
pip install numpy==1.18.5
# 或者
pip install numpy==1.18.5 -i https://pypi.tuna.tsinghua.edu.cn/simple
其中-i后面的网址,表示安装包的下载地址,国内pip安装速度慢,可以使用-i指定镜像源,加快安装速度
其他安装包也可以用pip一个一个的安装,或者直接安装所有依赖包:
pip install -r requirements.txt
PS:一般依赖包是向下兼容的,只需要安装比requirements版本号大于或者等于的版本即可
5. 常见的错误
- Windows环境中,项目不要出现中文路径,否则会出现opencv读取图片异常的问题
- 注意Windows和Linux的路径的分隔符
Windows路径分割符:【\】或者【//】 ,在python中某些库也可以使用【/】
Linux(Ubuntu)路径分割符号:【/】
- 如果出现“No module named ***”的错误,请使用pip install *** 即可,例如出现下面的错误
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'basetrainer'
请使用pip安装:
pip install basetrainer -i https://pypi.tuna.tsinghua.edu.cn/simple
其中-i后面的网址,表示安装包的下载地址,国内pip安装速度慢,可以使用-i指定镜像源,加快安装速度
相关文章
- python在windows和linux环境的进程对比
- 史上最详细 Python第三方库添加方法 and 错误解决方法
- python+selenium自动化测试 —— Windows环境搭建
- 【华为OD机试真题 python】 第k个排列【2022 Q4 | 100分】
- [windows菜鸟]C#中调用Windows API的技术要点说明
- 零基础教你快速入门Python怎么学python入门?python新手学习路线
- 掌握Python语言能做什么?python对小白友好吗?
- 零基础如何学Python?
- 中途转行python?怎么学?没有基础的我30了自学Python转行靠谱吗?
- python中定时任务apscheduler库用法详解
- 《Python数据挖掘:概念、方法与实践》一2.4 小结
- 【python】+704个常用工具Python库
- 【python】【pycharm】pyinstaller+python工程打包成exe/绿色免安装的文件夹+在windows下自动定时运行
- Python编程-数据库-利用PyMysql访问windows下的MySql数据库
- 《Python数据可视化编程实战》—— 1.5 在Windows上安装matplotlib
- Python 教程大全之使用 python 获取唯一单词的计数
- 【转载】 python进程绑定CPU
- Python 基础 之 Ubuntu 上安装 python 和 python-pip
- 【Leetcode】101:对称二叉树(Python)
- 【Pycharm技巧】:Pycharm删除多个不需要的python版本编译器
- 【python百度智能云】:Python — 三种获取__VIEWSTATE、__VIEWSTATEGENERATOR、__EVENTVALIDATION方法。
- 玩转Windows,超简单一个命令进行windows系统升级,在需要的时间点,随时进行系统升级,体验最新的系统和特性
- 【python】虚拟屏幕实现selenium截图的两种方法介绍
- Python系列之Windows环境安装配置
- python将h264文件视频转为mp4格式
- 计算机网络课程实验1——WINDOWS 环境下实现 WEB 服务器(python)