赞
踩
目录
为什么用PyTorch:简单来说,19年之前tensorflow是大哥,19年tensorflow和PyTorch双龙并行,20年之后PyTorch一往无前。宗旨,哪个用的人多用哪个。
1. 先打开Anaconda Prompt,创建虚拟环境
- conda create -n pytorch python=3.11.3 # 创建虚拟环境,python版本根据自己安装的可以更换
- conda info --envs # 检查创建的虚拟环境
- conda activate pytorch # 激活创建的虚拟环境
2. 打开pytorch官网
3. 如下图所示,找到Get Started按钮,按装选项可参考下图红色区域,固定选择PyTorch Build中Stable,Your OS中,根据电脑系统自行选择,Package中,推荐使用pip安装,最为省事,安装时如果没有pip3,可将代码中pip3改为pip,Language中,根据后面需要使用的语言安装,Compute Plaform中,GPU版本选用CUDA,CPU版本就选用CPU,最后网页会在Run this Command中自动生成代码,将该代码复制即可使用。
4. 安装成功之后,在虚拟环境中有些教程这样验证是否安装成功
- python
- import torch
- torch.cuda.is_available()
笔者尝试输出无果,遂用下列代码验证成功
- python
- import torch
- print(torch.__version__)
- print(torch.cuda.is_available())
注意:cpu版本只需要print(torch.__version__)就可以了,print(torch.cuda.is_available())是验证gpu版本的。
补充:离线安装
1.用cuda安装的用该代码卸载
- conda uninstall pytorch
- conda uninstall libtorch
2.用pip安装的用该代码卸载
pip uninstall torch
1.pytorch安装及卸载_conda 卸载pytorch_翟羽嚄的博客-CSDN博客
2.pytorch超详细安装教程,Anaconda、PyTorch和PyCharm整套安装流程_pytorch安装_李问号的博客-CSDN博客
3.Python安装教程步骤2:Windows中创建虚拟环境安装Pytorch并在PyCharm中配置虚拟环境_pycharm配置pytorch虚拟环境_ZHW_AI课题组的博客-CSDN博客
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。