非root用户安装cuda、pytorch、pyg,并使用ncu对pyg程序做分析

先看这个教程,在自己的目录里面安装cuda和cudnn。https://zhuanlan.zhihu.com/p/198161777  这个教程,写环境变量那几句,中间那一句应该是export PATH="/public/home/jd_lulin/cuda-10.1/bin:$PATH",而不是他正文中写的export PATH=$PATH:/home/zhaoqc/cuda-10.2/bin   注意啊,正文里面出现的路径、版本号,你要看情况替换成自己的。

安装完毕之后,如果nvcc -V显示cuda是你安装的那个版本,就成功了。然后就可以安装pytorch了。安装pytorch,是安装到虚拟环境中的,因此需要先创建一个虚拟环境,输入命令 conda create -n pyg python=3.8 就可以创建一个名为pyg的带有python3.8的虚拟环境。创建完毕之后,进入虚拟环境pyg,安装pytorch:输入命令conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch  然后就安装了(具体是什么命令,需要去pytorch官网的下载页面去看,找到你想要的版本,然后选择它对应的命令)。结束之后,在命令行输入python,然后输入import torch 看看有没有安装成功

然后安装pytorch geometric。https://blog.csdn.net/rothschild666/article/details/121696447?spm=1001.2014.3001.5506 看这篇博客就行

最后,在linux服务器上使用ncu报错:权限不足。因此使用如下命令:sudo ncu的绝对路径 -o 报告名称 python的绝对路径 xxx.py

猜你喜欢

转载自blog.csdn.net/illyh/article/details/131946835