大家可能习惯使用启动器一键安装,有时候在云端或使用的是官方包,没有启动器,这时候想升级就比较费劲。这是我升的方法,供大家参考。
安装torch+cu
1、使用进入torch的官方网站,选择想要安装的版本,默认为最新版251,点下方可以显示更多版本,如图所示
2、直接复制命令,贴入自己的命令行就可以了,如果是虚拟环境,可能需要带上自己的python路径。
注意,这里如果你已经安装了其它版本torch,那么系统会跳过,这里必须要指定你想要使用的版本
pip install torch==2.5.1+cu124 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
安装过程中可以看到后台信息,这里以云端为例
安装成功后,可以使用pip list
查看一下是否安装成功
3、之后需要升级一下xformers
配置的下载地址https://download.pytorch.org/whl/xformers/
我们可以直接用pip 升级
pip install -U xformers --index-url https://download.pytorch.org/whl/cu124
升级完成之后,重启环境即可。