1. Anaconda创建虚拟环境
- 在 Anaconda Prompt 中输入如下命令创建:
# pytorch-new 为你创建虚拟环境的名字
# python=3.8 为你指定的python 版本
conda create -n pytorch-new python=3.8
- 一路 yes即可。
2. 安装pytorch
根据自身电脑的英伟达硬件进行选择 pytorch版本
Pytorch安装地址
我的电脑是Windows,选择如下:
- 进入自己的虚拟环境
conda activate pytorch-newzc
- 在自己的虚拟环境下安装 pytorch & torchvision
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch
3. 小结
一定要用 anaconda 进入到自己的虚拟环境中安装自己的工具。