方法一:
pytorch官网:Previous PyTorch Versions | PyTorch
(推荐)我自己安装了一下环境,感觉这个教程不错:配置stylegan2环境遇到的一些问题 - 知乎
pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch/whl/lts/1.8/torch_lts.html -i https://pypi.tuna.tsinghua.edu/simple
或者安装可以匹配CUDA11的最低版本1.7.1,这样可能能适配一些老项目
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch/whl/torch_stable.html
CUDA的cuDNN在我的U盘上,当然你也可以在这个链接下载:
链接
cudnn-11.1-windows-x64-v8.0.4.30
链接:https://pan.baidu/s/1aqedvmiMXANVQqV5NPm7Yw
提取码:3ccj
cuda_11.1.0_456.43_win10
链接:https://pan.baidu/s/1KnhrYxptLQohVQ5Atql4sA
提取码:1td5
方法二
首先通过命令:安装python3.8的环境
conda create -n torch38 python=3.8
我的解决方法如下:
这里一定要注意python版本和cuda要对应,否则会错误
1、https://download.pytorch/whl/torch_stable.html
找到这个:cu111/torch-1.8.0%2Bcu111-cp38-cp38-win_amd64.whl
2、然后查看torch和torchvision对应关系要对应:
参考下面连接:
https://github/pytorch/vision#installation
3、这是下载完毕后的文件
在终端 pip install … (文件名).whl(这里要注意在该文件夹下安装,并且切换到你要安装的conda环境)
先安装torch
pip install "torch-1.8.0+cu111-cp38-cp38-win_amd64.whl" -i http://pypi.douban/simple --trusted-host pypi.douban
然后安装torchvision
即可安装完毕
这里我把这两个下载好的文件放在了U盘里,以后就可以直接安装了
Pytorch gpu(CUDA11.1)最新版本 官网路径 安装超时_氟小汀的博客-CSDN博客
更多推荐
CUDA+torch+vscode安装教程(python3.8)
发布评论