栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Linux cuda11.1安装torch

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Linux cuda11.1安装torch

  1. 创建虚拟环境
conda create -n torch18 python=3.8

进入虚拟环境

source activate torch18
  1. 安装torch1.8.1,可以从https://download.pytorch.org/whl/torch_stable.html中下载

    下载后直接 pip install whl文件即可

  2. 最关键的,下载torch_scatter,torch-sparse,torch-cluster,torch-spline-conv包,这些包依赖关系非常严重,下错了就是满屏红字。可以从https://pytorch-geometric.com/whl/中下载,找到对应的torch版本,例如我是torch1.8.1,则进入对应的目录

    出现下面的whl文件

下载下面的包(亲测有效)

torch_scatter-2.0.6-cp38-cp38-linux_x86_64.whl
torch_sparse-0.6.9-cp38-cp38-linux_x86_64.whl
torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl
torch_spline_conv-1.2.1-cp38-cp38-linux_x86_64.whl

下载完后直接pip install whl文件

最后直接pip install torch-geometric,不出意外,应该能顺利。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/998191.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号