目录
[1. 安装](#1. 安装)
[2. 卸载](#2. 卸载)
为什么用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 # 激活创建的虚拟环境 -
如下图所示,找到Get Started按钮,按装选项可参考下图红色区域,固定选择PyTorch Build中Stable,Your OS中,根据电脑系统自行选择,Package中,推荐使用pip安装,最为省事,安装时如果没有pip3,可将代码中pip3改为pip,Language中,根据后面需要使用的语言安装,Compute Plaform中,GPU版本选用CUDA,CPU版本就选用CPU,最后网页会在Run this Command中自动生成代码,将该代码复制即可使用。
-
安装成功之后,在虚拟环境中有些教程这样验证是否安装成功
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版本的。
2. 卸载
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博客