Pytorch win安装

1.确保conda已经换源，可参考

Conda 安装（Windows 系统）-CSDN博客

2.切换到自己创建的虚拟环境

查看环境信息

conda info -e

激活需要的环境

conda activate myenv

3.​​​​​​​Pytorch官网

PyTorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

输入y

下载成功

​​​​​​​4.验证

输入python之后

import torch

torch.cuda.is_available()

如果为false则不能使用GPU,建议下载显卡驱动

下载 NVIDIA 官方驱动 | NVIDIA

5.以后可能会使用到pip,建议更换源

pypi | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

如果您到 pip 默认源的网络连接较差，临时使用本镜像站来升级 pip：

python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip

配置

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple