pytorch和torchvision都是一起出现的,所以需要保证两者版本对应即可。

更新

更新其实就是将老版本进行卸载,再安装所需版本即可,其实卸载和安装都非常方便

卸载

直接使用下面命令对两者进行删除即可,也可以使用conda操作

pip uninstall torch
pip uninstall torchvision

安装

安装pytorch前必须保证安装了cuda和cudnn,这里推荐一个cuda的安装方法
https://www.cnblogs.com/arxive/p/11198420.html
安装最新的cuda即可。

运行cmd,输入nvcc --version来检查cuda是否安装成功以及查看cuda版本号。根据cuda版本来安装我们对应的pytorch和torchvision,我这里是10.2版本。
在这里插入图片描述

方法一:

官网安装,这里是官网地址 https://pytorch.org/,建议参考这篇博客先进行镜像换源成清华源https://blog.csdn.net/lyj223061/article/details/108639378,换源很快。
在这里插入图片描述
根据自己的需求获得安装命令,换源后输入命令即可快速安装。如果不换源会很慢,并且中途断开链接会直接导致安装失败。

方法二:

到这个地址https://download.pytorch.org/whl/torch_stable.html找到自己的安装版本,把torch和torchvision对应版本的whl文件直接下载到本地。pip install xxx.whl 安装两个包即可,亲测3秒就完事了,比第一种方法更快!

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐