pytorch2.1.0+cuda 12.1 环境配置自用学习笔记

pytorch2.1.0+cuda 12.1 环境配置笔记

1. Anaconda Powershell Prompt

conda env list  #查看现有虚拟环境
……
conda create -n python31011 python=3.10.11 #创建新的虚拟环境python3.10.11
……
conda activate python31011 #激活此新环境   退出（deactivate）
……
#库管理
pip list #查看已安装的库
pip install xxx #安装指定库

nvcc -v #查看cuda版本
nvida-smi #查看cuda信息
……
#测试
python
import torch
torch.cuda.is_available()
……

2.安装torch_geometric
第一步：打开torch_geometric官网https://pytorch-geometric.readthedocs.io/en/latest/install/installation.html
根据需要找到对应版本的安装命令
第二步：指定torch_geometric版本和文件名安装

首先搜索可用的版本
conda search pyg -c pyg #搜索可用的torch_geometric版本 -c 代表指定下载来源
……

找到自己需要的版本
执行命令进行安装：

conda install pyg=2.5.2=py310_torch_2.1.0_cu121 -c pyg

完成