- 安装Anaconda
官网下载安装 - 创建虚拟环境
打开Anaconda Prompt(开始菜单中)
创建虚拟环境,环境名为pytorch(可以任意起)
(1)创建环境pytorch,使用Python版本是3.7。
conda create -n pytorch python=3.7
之后,弹出提示,输入 y,即可安装。
(2)查看环境是否安装成功
conda info --envs
可以看到base和pytorch两个环境
(3)激活虚拟环境pytorch:activate pytorch
- 安装pytorch
安装前请到pytorch官网查看最新安装方法
https://pytorch.org/get-started/locally/
(1)输入安装命令
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
(2)将复制的代码粘贴到命令行格式下,弹出提示,输入 y,即可完成安装,显示“done”。
进入python:python
测试numpy:import numpy
测试pytorch:import torch
如果以上步骤都没问题,那么安装成功。
在虚拟环境中安装jupyter:conda install -y jupyter
虚拟环境中打开notebook :ipython notebook