windows上配置本地深度学习工作站

news/2024/11/22 20:08:10/

文章目录

      • 前言
      • 1. 安装Anaconda
      • 2. 创建python环境
      • 3. 安装Jetbrains Toolbox
      • 4. 安装Pycharm Professional/Community
      • 5. 安装cuda
      • 6. 安装cudnn
      • 7. 安装pytorch
      • 8. 配置环境变量
      • 9. 激活powershell的base环境
      • 总结

前言

搞深度学习两年了,所有代码都是在服务器上写的,导致到现在都没有一个本地深度学习工作站。这次碰上服务器坏了又坏,终于下定决心,在windows上配置本地深度学习工作站。整体配置顺序如下(*表示可选):

  • 安装Anaconda
  • 创建python环境
  • 安装Jetbrains Toolbox*
  • 安装Pycharm Professional/Community
  • 安装cuda
  • 安装cudnn
  • 安装pytorch*
  • 配置环境变量*
  • 激活powershell的base环境*

下面按照这个顺序简单介绍下

1. 安装Anaconda

按照官网给出的教程安装即可,过程中没有遇到什么bug。最后我把anaconda放在了D盘下,路径为D:\anaconda3

2. 创建python环境

打开Anaconda Prompt,这样打开后就直接进入到自动安装的base环境了

如果是打开cmd,那么首先要输入activate,这样才能激活base环境

可以选择直接用base环境,pip install需要的包就可以

也可以选择额外创建环境。这里使用指定路径的创建方式,方便之后定位:

conda create --prefix="D:\env\pt" python=3.6.3

然后激活:

conda activate D:\env\pt

然后再pip install需要的包

3. 安装Jetbrains Toolbox

直接在官网下载exe文件后再运行即可,安装过程非常快,很方便。可以选择语言为中文

此步骤是可选的,Toolbox就是集合Pycharm Community、Pycharm Professional等的工具包。根据我的使用习惯来说,先下载Toolbox再安装Pycharm更方便

4. 安装Pycharm Professional/Community

安装好Toolbox后,直接打开在Toolbox中点击安装Pycharm即可
在这里插入图片描述

这里可以选择Pycharm Community,但我还是觉得Professional更方便

5. 安装cuda

6. 安装cudnn

cuda和cudnn的安装参考知乎上一个教程,按照这个步骤下来也没遇到什么问题。需要注意的是,彼时cuda最新的版本已经是CUDA 11.8,然而pytorch可以适配的最高版本只是CUDA 11.6。目前来说,想搞深度学习的话,pytorch大概率是要用到的,所以在官网上选择了CUDA 11.6版本进行下载

这里插一句,由于我在安装cuda的过程中没有安装驱动,因此在查看cuda版本的时候,通过nvcc --versionnvidia-smi命令查看的版本是不一样的,前者显示11.6,后者显示11.4。至于原因是什么,可以参考这个博客

7. 安装pytorch

选好cuda版本,直接用官网给出的命令即可安装成功

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

很神奇,之前在服务器上,安装pytorch的命令总出问题,这次竟然一次成功

虽然安装成功,但后续在跑代码的过程中,报了AssertionError: Torch not compiled with CUDA enabled的错误,于是只能重新安装

最后我使用的是离线安装的手段,先在官网下载了cu113(至于为什么是cu113,是因为我的cuda版本显示了11.6和11.4两种,网上说cuda11.4时安装cu113的torch版本时成功的)的对应torch版本torch-1.10.0+cu113-cp36-cp36m-win_amd64.whl(使用cp36还是其他的是由电脑决定的,具体参考博客),然后跳转到下载位置目录,使用pip install --user ×××.whl命令,安装成功(如果下载的torch版本和本机不对应,会报错)。这里的离线下载参考了博客,报错参考了博客

8. 配置环境变量

这一步主要是为了下一步powershell可以自由激活python环境而准备的,参考csdn回答

9. 激活powershell的base环境

因为pycharm professional中的终端是powershell,所以需要额外搞这一步,如果觉得没用可以跳过。同样是参考csdn回答。配好之后就可以在powershell中通过activate命令激活base环境了,与cmd一样

总结

结束~整个过程大概会花费将近3h吧。整体来说配置过程还是很顺利的,后悔当初一直怕麻烦,没有搭建本地深度学习工作站。当然,windows上的显卡只是1650,运行很多项目还是太慢了,所以本地工作站主要用来写代码和简单的调试,真正跑实验和结果还是要去服务器。当然,也有比较两全其美的办法,就是个人拥有3090、4090之类的大显卡,目前本人正在筹备嘿嘿


http://www.ppmy.cn/news/148627.html

相关文章

探索时间与空间的奇妙舞蹈:Java 8中LocalDateTime的魔法时刻

引言 Java 8为开发人员引入了众多强大的新特性和改进,其中之一就是LocalDateTime类。LocalDateTime是一个日期和时间的对象,它不包含时区信息。本文将详细介绍Java 8中LocalDateTime的用法,并提供示例来说明其功能和用途。 一、LocalDateTi…

MyBatis - 动态 SQL

文章目录 1.if2.choose3.where4.set5.foreach6.基于注解编写 动态 SQL 是 MyBatis 提供的一个非常强大的功能,它可以让我们在运行时构建 SQL 语句。这意味着我们可以根据应用程序的需求来构建符合要求的 SQL。通常情况下,这是非常有用的,因为…

Android 12系统源码_窗口管理(四)RootWindowContainer简介

前言 在Android 12中,RootWindowContainer是一个重要的类,它是WindowManagerService的主要组成部分之一,用于管理所有应用程序的窗口。它表示整个屏幕上的最顶层的容器,包括状态栏、导航栏和所有应用的窗口。 一、RootWindowContainer的创建 1、RootWindowContainer是在…

tcpdump 抓包工具详细图文教程(上)

目录 一、tcpdump 抓包工具的基本介绍和学习基础 1.1 常用的抓包工具 1.2 tcpdump 抓包工具介绍 二、tcpdump 抓包工具使用环境和初体验 2.1 编译安装 tcpdump 2.2 抓包 三、讲解 TCP 协议报文报头 四、tcpdump 抓包工具常规过滤规则 4.1 tcpdump 的 host 和 net 过…

MVCC 实现原理

💟这里是CS大白话专场,让枯燥的学习变得有趣! 💟没有对象不要怕,我们new一个出来,每天对ta说不尽情话! 💟好记性不如烂键盘,自己总结不如收藏别人! &#x1f…

佳能EF镜头控制器

机器视觉应用相机通常选择SLR镜头,牺牲了控制控制功能,所以不能用于远程控制。使用LC-2镜头控制系统搭配佳能(Canon)EF镜头可以配合机器视觉相机使用从而实现远程控制。 EF/EFS 镜头控制单元(LC-2) 适用于大部分佳能(Canon) EF和腾龙(Tamron) EF镜头 精…

什么镜头最适合拍风景_要拍风景和人像,分别买什么镜头比较好?

展开全部 没有固定的用法。 但是人像基本用特写的手法,那么中长焦镜头是e5a48de588b63231313335323631343130323136353331333365643632最好的选择,大光圈容易带来虚化的效果,更能突显人物,所以在摄影镜头中85mm和135mm两个经典定焦…

佳能7款PL卡口电影镜头发布在即

本文来自新摄影 在四月份NAB展会之前,佳能将发布7款PL卡口电影镜头。与此前的EF卡口电影镜头相比,这几款PL卡口镜头的光学设计略有不同,且拍摄出来的画面更加柔和、拥有更低的对比度。据了解,目前佳能只有6款CN-E电影变焦镜头镜头…