针对cuda11.1版本安装pytorch-gpu版本@TOC
配了3天的环境终于true了,简单列举一下这三天遇到的问题和解决方法。
首先介绍一下成功的路径。
由于校园网不好,我使用了离线下载。
在下载torch之前,最好先打开命令行(输入cmd),在命令行中输入nvidia-smi查看显卡对应的版本:
然后打开这个网址https://download.pytorch.org/whl/torch_stable.html,寻找版本对应的torch和torchvision进行下载。
此处介绍gpu版本,因为有英伟达显卡,所以选择下载gpu版本,此处我下载的是这两个
cu111/torch-1.9.0%2Bcu111-cp38-cp38-win_amd64.whl
cu111/torchvision-0.10.0%2Bcu111-cp38-cp38-win_amd64.whl
(cu111是对应cuda11.1),选择windows版本而不是linux,因为这里我有遇到过,先下载了linux版本然后报错,因为文件里只能识别win_amd64类似的名字。所以要选对版本。
我是将它们下载到了D盘的安装包里,然后在anaconda prompt中首先要进入存放这两个包的地址中,然后新建虚拟地址