前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >深度学习Pytorch(一)

深度学习Pytorch(一)

原创
作者头像
用户8178468
修改2023-03-09 15:28:54
3400
修改2023-03-09 15:28:54
举报
文章被收录于专栏:bgemini的文章bgemini的文章

深度学习Pytorch(一)

前言:必须使用英伟达显卡才能使用cuda(显卡加速)!

移除环境:

代码语言:shell
复制
conda remove -n pytorch --all

一、安装Pytorch

  1. 下载Anaconda
  2. 打开Anaconda Prompt
  3. 创建一个Pytorch环境:
代码语言:javascript
复制
conda create -n pytorch python=3.9
  1. 激活Pytorch环境:
代码语言:javascript
复制
 conda activate pytorch
  1. 查看当前包:
代码语言:javascript
复制
 pip list
  1. 安装Pytorch
代码语言:javascript
复制
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
  1. 查看torch是否安装成功
代码语言:javascript
复制
python
import torch
torch.cuda.is_available()

如果显示true说明torch可以使用咱们的英伟达显卡!

二、配置Pycharm

  1. 打开Pycharm,创建新项目

导入我们刚才创建的Pytorch环境的Python编译器

测试是否可以导入使用conda,打开Python控制台

代码语言:javascript
复制
 import torch
 torch.cuda.is_available()

出现True说明配置成功!

三、配置Jupyter

因为最开始安装的Anaconda中的Jupyter只存在于base环境中,无法在我们新建的Pytorch环境 中使用,所以我们需要进行以下操作在新环境中安装Jupyter

进入Pytorch环境中,需要安装一个包:

代码语言:javascript
复制
conda install nb_conda

安装完成后打开Jupyter:

代码语言:javascript
复制
jupyter notebook

打开后会弹出一个网页,使用Pytorch环境的Python新建一个代码:

输入以下代码进行测试:

代码语言:javascript
复制
import torch
torch.cuda.is_available()

输入完一行代码后,按Shift+Enter执行!出现上图说明配置成功!

四、Python学习中的两大法宝(也可以用在Pytorch中)

以Pytorch包举例我们想了解一个包的构成和使用方法:

代码语言:txt
复制
dir()   //打开,看见
help()   //说明书
比如:
dir(Pytorch)
输出:1、2、3、4

dir(Pytorch.3)
输出:a,b,c

help(Pytorch.3.a)
输出:将此扳手放在特定的地方,然后拧动

总结:dir()函数,能让我们知道工具箱以及工具箱中的分隔区有什么东西。

代码语言:txt
复制
help()函数,能让我们知道每个工具是如何使用的,工具的使用方法。
代码语言:txt
复制
具体使用help()函数时,指定到工具不加(),例如help(torch.cuda.is_available)

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 深度学习Pytorch(一)
    • 一、安装Pytorch
      • 二、配置Pycharm
        • 三、配置Jupyter
          • 四、Python学习中的两大法宝(也可以用在Pytorch中)
          领券
          问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档