如果之前安装过CUDA的其他版本,并且安装了VS Studio,安装时选择自定义安装,将11.6的有关VS的组件全部取消安装。虽然CUDA能安装成功,但是VS能不能本人还没验证过,不过pytorch相关的包全部安装成功
官方地址
# CUDA 11.6
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
其中 ${CUDA} 和 ${TORCH} 应分别替换为特定的 CUDA 版本(cpu、cu102、cu113、cu115)和 PyTorch 版本(1.11.0、1.12.0)。 例如,对于 PyTorch 1.12.* 和 CUDA 11.6,输入:
pip install torch-scatter -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric
再选择安装
pip install torch-cluster -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html