【win10】安装pytorch1.6.0+cuda10.1 + torch-geometric

conda下新建一个环境,使用python3.6

conda create --name torch_ge python=3.6

安装pytorch1.6.0

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

安装cuda10.1+cudnn

参考,
百度云链接:https://pan.baidu.com/s/1HqyGjha1IkS6vBFlzxQ82g
提取码:ls3d

安装torch-geometric

pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-sparse==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-cluster==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-spline-conv==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-geometric

你可能感兴趣的:(其它,pytorch,深度学习,神经网络)