cuda118安装torch
时间: 2023-09-14 16:03:36 浏览: 306
cuda cudnn torch torchvision 安装包
要在CUDA 11.1.0上安装torch,您可以使用以下命令:
pip install torch==1.8.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
请确保您的CUDA版本与所安装的torch版本相匹配。另外,还要注意依赖关系,确保您已经安装了torch_scatter、torch_sparse、torch_cluster和torch_spline_conv这些包。您可以从https://pytorch-geometric.com/whl/下载这些包,并确保选择与您的torch版本匹配的正确文件进行安装。<span class="em">1</span><span class="em">2</span><span class="em">3</span><span class="em">4</span>
阅读全文