Linux完全卸载PyTorch&重装(cuda11.1)
原因:网上看的很多卸载的,并没有把所有跟随PyTorch的包卸载干净,虽然可能后期不会出现较大问题,但本着干干净净,把所有带来的都删了。
·
原因:网上看的很多卸载的,并没有把所有跟随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.1
pip 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
出现以下界面,安装配置成功!
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
已为社区贡献1条内容
所有评论(0)