pytorch和torchvision安装、卸载、版本查看

一、安装pytorch和torchvision

1、配置清华镜像源

我们使用Anaconda环境进行Pytorch的安装
进入到Anaconda环境后,在命令行中依次输入一下命令并执行

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

2、pytorch官网下载相应版本即可

https://pytorch.org/get-started/previous-versions/
注意删除命令中的-c pytorch,因为加了这个就是从官网下载,速递较慢

二、卸载pytorch和torchvision

pip uninstall torch
pip uninstall torchvision

三、查看pytorch和torchvision版本

打开终端,如果安装在虚拟环境中,切换进虚拟环境中
输入:python
进入python环境中,依次输入:

import torch
torch.__version__
import torchvision
torchvision.__version__

如下所示:
在这里插入图片描述

### PyTorch镜像源安装配置 #### 使用清华镜像源安装PyTorch 为了加速下载速度以及提高稳定性,在Linux系统上推荐使用清华大学开源软件镜像站来安装PyTorch。具体操作如下: 首先更新`conda`的渠道设置,添加清华镜像作为优先级最高的频道[^4]。 ```bash conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ ``` 接着可以验证当前使用的镜像源是否已更改为清华镜像源: ```bash conda config --show | grep -A3 channels ``` 对于特定版本CUDA的支持, 查看本地CUDA版本号之后再决定要安装哪个版本PyTorch。如果已经确认好所需的CUDA版本,则可以直接通过以下命令完成安装[^2]: ```bash conda install pytorch torchvision torchaudio cudatoolkit=XX.X -c pytorch ``` 这里将`XX.X`替换为实际需要的CUDA版本号即可。 另外一种方式是利用`.whl`文件来进行离线安装或针对Python不同版本进行精确控制。例如在Lubuntu环境下可以通过pip工具配合具体的wheel文件名实现快速部署[^1]: ```bash pip install torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl ``` 需要注意的是上述例子中的轮子文件路径名称需根据实际情况调整。 最后提醒一点关于环境管理方面的小技巧:当不再需要某个特定版本PyTorch开发环境时,可通过`conda env remove`指令轻松移除整个虚拟环境以释放资源[^3]。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

L-Dreams

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值