深度学习
Windows下开发环境搭建
2. Anaconda环境安装
2.6 Anaconda安装Pytorch
官网：https://pytorch.org/get-started/locally/
- 创建pytorch环境
conda create --name pytorch python=3.8
- 激活pytorch环境
conda activate pytorch
- 添加pytorch镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
- 安装pytorch，在官网选择版本，复制命令，注意：去掉
-c pytorch，会快一点。
conda install pytorch torchvision torchaudio cudatoolkit=11.0
- 看看是否安装成功，是的，安装成功！
python
import torch
torch.__version__
torch.Tensor([1,]).cuda()
- OK，到这里，Pytorch就安装好了。
补充：
- PyTorch中安装matplotlib
conda activate pytorch
python -m pip install -U matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
创作不易，喜欢的话加个关注点个赞，❤谢谢谢谢❤
转载：https://blog.csdn.net/xu_benjamin/article/details/117528897
查看评论