跨境派

跨境派

跨境派,专注跨境行业新闻资讯、跨境电商知识分享!

当前位置:首页 > 工具系统 > 运营工具 > Linux完全卸载PyTorch&重装(cuda11.1)

Linux完全卸载PyTorch&重装(cuda11.1)

时间:2024-04-19 11:15:22 来源:网络cs 作者:往北 栏目:运营工具 阅读:

标签:

目录

1.命令行输入2.命令行接着输入3.可以看到卸载完后4.下载PyTorch(cuda对应11.1,可以通过test_gpu.py的测试)5.测试pyTorch是否在使用GPU
原因:网上看的很多卸载的,并没有把所有跟随PyTorch的包卸载干净,虽然可能后期不会出现较大问题,但本着干干净净,把所有带来的都删了。

1.命令行输入

conda uninstall pytorch

在这里插入图片描述

2.命令行接着输入

conda uninstall pytorch-cuda(这一步许多教程中没有,导致还有十多个包没删除)

在这里插入图片描述

3.可以看到卸载完后

查看包的命令:conda list

在这里插入图片描述
之前跟随PyTorch的包都没了
在这里插入图片描述

4.下载PyTorch(cuda对应11.1,可以通过test_gpu.py的测试)

PyTorch官网【https://pytorch.org/get-started/previous-versions/】给出的命令:

# CUDA 11.1pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述
或者:

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述

5.测试pyTorch是否在使用GPU

出现以下界面,安装配置成功!

在这里插入图片描述

本文链接:https://www.kjpai.cn/news/2024-04-19/160287.html,文章来源:网络cs,作者:往北,版权归作者所有,如需转载请注明来源和作者,否则将追究法律责任!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。

文章评论