pytorch安装步骤

1.找到相应版本

• 版本连接查找
  • https://github.com/pytorch/vision#installation
• 查看环境命令
  • https://pytorch.org/
    

2.下载步骤cpu版本

2.1选择下载版本

• torch 1.12.0
• torchvision 0.13.0
• Python 3.7

2.2 创建虚拟环境

conda create -n env_torch python==3.7

2.3 进入虚拟环境

conda activate env_torch 

2.4 安装pytorch

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio cpuonly -c pytorch

2.5 验证是否安装成功

import torch
torch.__version__

2.6 安装ipykernel

 conda install ipykernel

2.7 将新建的环境环境写入notebook的kernel中

python -m ipykernel install --user --name env_torch --display-name “env_torch”

2.8 在虚拟环境中打开jupyter

jupyter notebook

2.9 选择创建的虚拟环境创建文件

2.10 jupyter 中测试pytorch

import torch
torch.__version__
x = torch.rand(5,3)
x

完成