C++/CLI support
和VC++ 2015.3 v14.00(v140) toolset for Desktop
两个在visual C+++ build tools下面的包conda install PyTorch=1.2 cudatoolkit=10.0 torchvision -c pytorch
。这里需要注意,最好让cudatoolkit的版本和第二步安装的CUDA的版本是 相同 。
第四步是pip install --verbose,分别以下几个package:
1. torch-scatter
2. torch-cluster
3. torch-sparse
这几个包都需要nvcc编译,每个包在编译过程中都有多个.cu
文件,安装过程中的错误可能来自:
cl.exe
的路径选择不对,在系统环境变量PATH中,是添加了D:\Program Files (x86)\VisualStudio\VC\Tools\MSVC\14.16.27023\bin\Hostx64\x64
,一般在命令行中输入cl
会有输出。不要选择arm64和x86版本。同时要把14.16.27023\bin\
下的Hostx86
等文件夹删掉,不然在我编译torch-sparse
的过程中,在PATH
配置正确的情况下,竟然还有调用Hostx86
下的cl.exe
。--verbose
选项,直接pip install torch-scatter
,或者直接安装了没有正确编译的whl缓存,需要注意一下安装过程的输出信息,如果有出现install cache ..../..../.../torch-scatter.whl
的情况,那么还得把whl缓存删掉。第五步是pip install --verbose,分别是以下几个package:
1. pylfile
2. rdflib
pip install --verbose torch-geometric
Win10 下安装PyTorch 1.2 (GPU)和PyTorch-geometric (PyG)的记录
原文:https://www.cnblogs.com/Tancred-kPro/p/11630999.html