torch安装教程

1、首先确定机器使用的cuda版本nvcc -V

例如本服务器使用的时cuda11.1版本

2、创建一个conda环境:

conda create -n new_env python=3.8

conda activate new_env

3、安装pytroch环境:
conda install pyg -c pyg
该命令会把一些常用的torch以及关联的工具包都安装到机器,验证是否正确安装:

如果发生错误:

需要自行去官网查找正确的版本:Previous PyTorch Versions | PyTorch

4、安装torch-geometric:
torch-geometric要和torch版本配套,登录网站链接: 链接查看对应的版本:

最后安装torch-geometricpip install torch-geometric
失败了可以尝试加上版本号pip install torch-geometric==2.0.4

5、验证torch-geometric:

成功导入即可

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2023年4月27日
下一篇 2023年4月27日

相关推荐