TorchEEG

install Pytorch

refer to https://pytorch.org/

check torch version //v1.11

print(torch.version) python -c "import torch; print(torch.version)"

check CUDA verison //v11.6

cd C:\Program Files\NVIDIA Corporation\NVSMI

nvidia-smi.exe

install pytorch-geometric

refer to https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.11.0+cu115.html

Last updated

Was this helpful?