Win10+cuda10.1+pytorch1.3+torch_geometric成功安装

经过多次反复终于安装成功,记录下来完整步骤如下:

一、安装visual studio2017

1.Visual Studio 2017 Community(必须安装,Cuda是与其结合使用的)

下载地址:https://visualstudio.microsoft.com/zh-hans/free-developer-offers/

安装选项:勾选“C++的桌面开发”即可。
2.勾选了“用于桌面的VC++2015.3 v140工具集(x86,x64)”和“C++/CLI支持”

二、安装cuda

1.安装cuda

首先需要查看自己电脑的N卡支持的 CUDA 版本,打开 NVIDIA 控制面板——帮助——系统信息——组件:

在这里插入图片描述
接下来参照https://blog.csdn.net/u012319441/article/details/95551632

需要添加一波环境变量

$ CUDA_PATH: C:\Program Files\NVIDIA GPU
Computing Toolkit\CUDA\v10.1

$ CUDA_PATH_V10_1: C:\Program Files\NVIDIA
GPU Computing Toolkit\CUDA\v10.1

$ CUDA_BIN_PATH: %CUDA_PATH%\bin

$ CUDA_LIB_PATH: %CUDA_PATH%\lib\x64

$ CUDA_SDK_PATH: C:\ProgramData\NVIDIA
Corporation\CUDA Samples\v10.1

$ CUDA_SDK_BIN_PATH:
%CUDA_SDK_PATH%\bin\win64

$ CUDA_SDK_LIB_PATH:
%CUDA_SDK_PATH%\common\lib\x64

然后在 系统变量 的 Path —— 新建,添加四条信息

$ %CUDA_BIN_PATH%

$ %CUDA_LIB_PATH%

$ %CUDA_SDK_BIN_PATH%

$ %CUDA_SDK_LIB_PATH%

2. 安装 CUDNN

(1)选择与CUDA 版本相匹配的 cuDNN 版本。在下载时需要先在官网进行注册

下载的压缩包:cudnn-10.1-windows10-x64-v7.6.4.38.zip

(2)将下载文件解压缩,然后把里面 bin、include 以及 lib 文件夹中的文件分别复制到 CUDA 安装目录 的 bin、include 以及 lib 文件夹下面。

打开命令提示符,输入:nvcc -V

出现如下类似信息即为成功
在这里插入图片描述

三、安装anaconda

在官网https://www.anaconda.com/distribution/下载,这里用的版本是

Anaconda3-2019.10-Windows-x86_64.exe

参照博客:https://blog.csdn.net/chengyuqiang/article/details/101757019

一路回车安装就可以了。

四、安装pytorch

登录官网https://pytorch.org/get-started/locally/,进行如下选择
在这里插入图片描述
然后执行命令安装:pip3 install torch=1.3.1 torchvision=0.4.2
-f https://download.pytorch.org/whl/torch_stable.html

安装成功后可以看到:
在这里插入图片描述

五、安装pytorch_geometric

https://github.com/rusty1s/pytorch_geometric

依次安装下列包:

$ pip install --verbose --no-cache-dir torch-scatter

$ pip install --verbose --no-cache-dir torch-sparse

$ pip install --verbose --no-cache-dir torch-cluster

$ pip install --verbose --no-cache-dir
torch-spline-conv (optional)

$ pip install torch-geometric

安装成功:
在这里插入图片描述
[1]:https://blog.csdn.net/u012319441/article/details/95551632
[2]:https://blog.csdn.net/chengyuqiang/article/details/101757019
[3]:https://blog.csdn.net/p312011150/article/details/81571273

  • 5
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值