Pytorch安装与使用

668 阅读1分钟

一、安装Anaconda

Anaconda安装/jupyter安装与使用

二、安装pytorch

1 打开Anaconda prompt,创建pytorch环境

输入 conda create -n pytorch python=3.7
等待。。。 1.创建pytorch环境.png

运行结束,出现proceed?
输入y
等待。。。 2.png

运行结束,创建pytorch环境成功 3.安装成功.png

2 安装pytorch

输入nvidia-smi 查看支持版本信息 4.CUDA版本选择.png

去PyTorch官网下载支持版本 5.CUDA下载.png

进入pytorch环境:
输入 conda activate pytorch 7.切换pytorch环境.png

复制相应版本安装信息:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch 6.安装.png

3 验证

通过导包验证安装是否成功
import torch
不报错就是安装成功 8.验证安装成功.png

验证pytorch是否可以使用本机GPU
输入 torch.cuda.is_available() 9.验证是否能够是哟gpu.png

三、安装pycharm

Pycharm安装与使用

四、Pycharm配置pytorch环境

1 创建工程,选择环境

10.pycharm配置pytorch.png

2 测试

11.pycharm配置测试.png

五、jupyter使用pytorch

在安装Anaconda时,会附带直接安装了jupyter notebook,但是jupyter默认是在base环境中的,所以jupyter是无法直接使用pytorch的。
两种解决办法:
    一是在base环境中在安装pytorch
    二是在pytorch环境中安装jupyter(推荐)

1 在pytorch环境中安装jupyter

打开Anaconda prompt,切换环境到pytorch,安装: 12.pytorch中安装jupyter.png

打开jupyter notebook: 13.运行jupyter.png

jupyter notebook中测试是否可用: 14.jupyter中测试.png