文章目录
- cuda 安装
- PyTorch 1.x版本安装
cuda 安装
官网:cuda各个版本安装教程
选择相应版本点击版本前方链接就可以进入安装教程页面
例如:我想要为ubuntu系统安装一个11.7版本的cuda,则选择11.7版本的连接,然后进入安装教程页面。
然后依次选择对应的选项,最后就会出现安装的命令,如果是第一安装的话,最下方的那种方式都是可以的。如果不是第一次安装的话,建议选择runfile
的方式进行安装,可以省去一些小麻烦。
安装命令会出现在下方,例如:
例如,11.7版本的就是:
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
sudo sh cuda_11.7.0_515.43.04_linux.run
直接复制粘贴到命令行就可以安装了。
安装完毕之后,可以清理掉下载的原始文件,节约空间。
安装完成之后,cuda版本应该还是之前的版本,需要进行版本切换。
直接设置环境变量,将变量指向需要的cuda
版本目录即可。
安装路径在:
/usr/local/
下
PyTorch 1.x版本安装
官网对应安装命令
需要注意与cuda的版本对应的关系,选择好之后,选择一种适合自己的直接方式安装就好。
提醒自己:
遇到问题需要冷静,愤怒不会让你变强,反而会让你的效率更慢。心平气和的慢慢来,可能反而会更快。
加油!