经过多次反复终于安装成功,记录下来完整步骤如下:
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支持”
首先需要查看自己电脑的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%
(1)选择与CUDA 版本相匹配的 cuDNN 版本。在下载时需要先在官网进行注册
下载的压缩包:cudnn-10.1-windows10-x64-v7.6.4.38.zip
(2)将下载文件解压缩,然后把里面 bin、include 以及 lib 文件夹中的文件分别复制到 CUDA 安装目录 的 bin、include 以及 lib 文件夹下面。
打开命令提示符,输入:nvcc -V
在官网https://www.anaconda.com/distribution/下载,这里用的版本是
Anaconda3-2019.10-Windows-x86_64.exe
参照博客:https://blog.csdn.net/chengyuqiang/article/details/101757019
一路回车安装就可以了。
登录官网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
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