首先在pytorch 官网下载对应版本的.whl文件
官网链接:https://download.pytorch.org/whl/torch_stable.html
找到自己需要的版本,下载需要十几分钟。
下载完成后,打开anaconda prompt,进入虚拟环境,这里以名为“pipelearn”的环境为例,
conda activate pipelearn
使用pip安装,我的whl文件名称为torch-2.0.0+cu118-cp39-cp39-win_amd64.whl
pip install torch-2.0.0+cu118-cp39-cp39-win_amd64.whl
安装成功后如图
安装torchvision
使用pip安装,这里要注意默认的安装为最新版本,这会导致上一步安装的torch自动更新,可能会引起与gpu版本不一致,所以需要指定torchvision的安装版本。版本对应如下
以我需要的版本为例
pip install torchvision==0.15.1
最后查看是否安装完成
conda list