pytorch学习笔记环境的安装以及工具的选择和使用

最新推荐文章于 2024-03-19 10:01:07 发布

原创最新推荐文章于 2024-03-19 10:01:07 发布 · 150 阅读

CC 4.0 BY-SA版权

本文介绍了如何安装Anaconda，并在其中创建一个名为pytorch的Python 3.6虚拟环境。接着，详细阐述了如何在该环境中安装PyTorch、检验安装成功以及打开和使用Jupyter Notebook。此外，还提到了在PyCharm中配置PyTorch环境的方法，并介绍了dir()和help()两个实用函数的用途。

部署运行你感兴趣的模型镜像

安装anaconda环境的步骤：

首先先安装anaconda
https://repo.anaconda.com/archive/这是不同版本的地址

在anaconda里创建一个虚拟环境
conda create -n pytorch python=3.6
（ 一个叫pytorch的python3.6的环境）

安装pytorch 
进入官网https://pytorch.org/

在这里插入图片描述

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c 				pytorch -c conda-forge   
在终端输入这个命令进行安装pytorch

检验是否安装成功
输入python是进入环境
再输入import torch如果没报错就是成功的
再输入torch.cuda.is_available() 返回的true表示pytorch可以使用gpu

怎么打开jupyter

点击anaconda进入终端
输入conda activate pytorch进入虚拟环境
在输入jupyter notebook就可以打开了

jupyter和pycharm的使用

因为pytorch环境里没有jupyter所以我们要在pytorch环境里安装jupyter

安装步骤
安装jupyter note book
进入pytorch环境以后 输入conda install nb_conda安装
完成后输入jupyter notebook 就会直接进入jupyter的首页 
然后点击右上角的new 
就可以创建一个项目了 
检测有没有成功安装的方法是 
先import torch 然后按shift+enter
再输入torch.cuda.is_available()
返回true就表示成功了

pycharm里面的操作

创建一个新的项目
环境的选择如下所示

两个很有用的函数

dir()函数 让我们知道这个函数里有哪些东西

help()函数 让我们知道怎么使用一个工具，以及一个工具的作用

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理