pytorch + cuda +pyg 安装
1. conda 虚拟环境安装
查看所有虚拟环境 conda env list
删除指定虚拟环境 conda env remove -name yor_env_nam
2. 安装pytorch + cuda
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
pytorch网站:https://pytorch.org/get-started/locally
3.安装pyg
进这个网站 分别安装指定版本的包
https://data.pyg.org/whl/torch-1.12.0%2Bcu113.html pip install torch-geometric
4 .内核
添加内核
conda install nb_conda_kernels # 安装nb_conda_kernels conda install ipykernel # 安装ipykernel # 将环境写入jupyter notebook的kernel中 python -m ipykernel install --user --name env_name --display-name "你想为kernel添加的名称"
删除内核 jupyter kernelspec remove env_name # 如果给env_name起了别名,将env_name替换成别名