admin管理员组

文章数量:1564178

目录

    • 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/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/whl/torch_stable.html


或者:

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

5.测试pyTorch是否在使用GPU

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

本文标签: 重装PytorchLinuxamp