PyTorch 安装

  • 本机环境

打开设备管理器,找到相应的显卡,查看本机是否支持GPU,如下图所示,这可以确定需要安装PyTorch 是GPU 还是CPU 版本

  • Anaconda 安装

Anaconda官网地址:https://www.anaconda.com/distribution/,选择和机器环境对应的版本并下载安装

 

  • PyTorch 安装

PyTorch 主页:https://pytorch.org/

PyTorch 主要使用Pip 或 Conda安装,这里使用Anaconda作为安装工具,选择CUDA版本

 

打开Anaconda,创建一个新的环境,指定Python的版本

 

 

 

 

打开Anaconda Prompt,可以查看当前的环境或者切换环境:

 

 

 

 

列出环境:conda info --env

切换环境:conda activate My_PyTorch

查看安装的包:conda list

 

使用Pip方式安装,命令:pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113,执行过程如下

 

  • 验证PyTorch是否安装成功

1:打开python:python

2:测试pytorch:import torch

如下图所示,没有遇到问题,则表示成功了

 

 

  • 安装Jupyter

安装Anaconda的同时,会在base环境安装jupyter notebook,因此并不能在此jupyter notebook运行torch包,在前面新建的环境中运行命令

conda config --set remote_read_timeout_secs 1000.0

conda install nb_conda

 

 

完成后,执行jupyter notebook,如下图,选择之前创建的conda 环境

 

输入import torch,点击运行
再输入torch.cuda.is_available()
输出 True,表示我们在自己创建的pytorch环境中安装好了jupyter包

 

posted @ 2022-09-06 10:09  黎明踏浪号  阅读(492)  评论(0编辑  收藏  举报