win10安装cuda10.1+cudnn7.6.4+pytorch1.4.0及torch_geometric

cuda10.1安装

相比linux安装要简单很多,打开https://developer.nvidia.com/cuda-downloads
win10安装cuda10.1+cudnn7.6.4+pytorch1.4.0及torch_geometric_第1张图片点击蓝框
win10安装cuda10.1+cudnn7.6.4+pytorch1.4.0及torch_geometric_第2张图片进行下载,这个下载很快。
https://docs.nvidia.com/cuda/archive/10.1/cuda-installation-guide-microsoft-windows/index.html这是英伟达官网的下载文档说明,如果想万无一失的下载,就好好看看这个。

CUDNN下载

https://developer.nvidia.com/rdp/cudnn-archive
在此链接下下载CUDNN

win10安装cuda10.1+cudnn7.6.4+pytorch1.4.0及torch_geometric_第3张图片win10选这个。
现在cudn与cudnn全部下载下来了
输入nvcc -V查看是否安装成功
win10安装cuda10.1+cudnn7.6.4+pytorch1.4.0及torch_geometric_第4张图片

pytouch下载与安装

https://pytorch.org/get-started/previous-versions/进入该链接
win10安装cuda10.1+cudnn7.6.4+pytorch1.4.0及torch_geometric_第5张图片直接在anaconda里面输入该指令

conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch

然后就开始下载了
win10安装cuda10.1+cudnn7.6.4+pytorch1.4.0及torch_geometric_第6张图片记得中间点一个yes,网速比较慢
如果等不了的话
就用镜像,先添加下载路径(分别打进去)
(参考https://blog.csdn.net/qq_34037389/article/details/105083960)

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

然后执行
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1
pytorch就安装好了

安装五个小弟

https://pytorch-geometric.com/whl/torch-1.4.0.html
pytorch安装好了,但是还有4个配件用于图神经网络库PyTorch geometric,可以参考
https://zhuanlan.zhihu.com/p/138826540/
从这个链接下载
https://pytorch-geometric.com/whl/torch-1.4.0.html
分别是
torch_cluster-1.5.3-cp36-cp36m-linux_x86_64.whl
torch_scatter-2.0.4-cp36-cp36m-linux_x86_64.whl
torch_sparse-0.6.0+cu101-cp36-cp36m-linux_x86_64.whl
torch_spline_conv-1.2.0-cp36-cp36m-linux_x86_64.whl
下载下来之后用pip安装,注意一定要安装在你的虚拟环境下。
最后

pip install torch_geometric

这个时候应该就ok了

你可能感兴趣的:(pytorch,windows,10)