1.Anacond安装
下载官网:https://www.anaconda.com/
1.1点击next
1.2点击 i agree
1.3点击next
1.4点击next
*
1.5安装结束后 启动
通过conda创建一个名为:pytorch的虚拟环境
conda create -n pytorch python=3.9.6
1.7查看创建安装的环境:
conda info --envs
安装pytorch
进入pytorch的官网
https://pytorch.org/
GPU安装CUDA下载
进入自己的pytorch环境
conda activate pytorch
在输入
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
验证
import torch
torch.cuda.is_available()
删除环境
conda remove -n pytorch --all
pytorch 是要删除的环境名