整体参考:https://blog.csdn.net/Yakusha/article/details/144306858
安装完成后的整体版本信息
- ubuntu:22.04
- 内核:6.8.0-51-generic
- 显卡:NVIDIA GeForce RTX 3050 OEM
- 显卡驱动:535.216.03
- cuda:12.2
- cudnn:8.9.7
- python:3.8
- pytorch:2.4.1+cu121
一、安装nvidia显卡驱动
1、安装驱动
- 查看现在是否安英伟达的显卡驱动
nvidia-smi
- 如果安装成功就会像下面这样,可以看到现在的驱动是535,cuda最高支持12.2
- 如果没有安装直接用 “附加驱动” 图形安装就可以了,现在图形界面安装都比较成熟了。
- 这里踩到的一个坑是之前用的内核是加了实时补丁包的实时内核版本,但是后来发现实时内核和nvidia驱动冲突,所以实时内核按不上显卡驱动。
- 更新显卡驱动后,网没了,原因是内核更新了。
解决方法:https://blog.csdn.net/weixin_44286126/article/details/131455726
2、安装cuda+cudnn
cuda安装:https://blog.csdn.net/str1ngthen/article/details/133793250
cuda下载地址:https://developer.nvidia.com/cuda-12-2-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=runfile_local
cudnn安装:https://blog.csdn.net/tyyhmtyyhm/article/details/136840299
cudnn下载地址:https://developer.nvidia.com/cudnn-archive
- 这部分安装照着教程走就可以了,但是要注意版本对应。
- 生成命令,执行即可
- 下载tar压缩形式,解压后存到cuda对应地址
二、安装miniconda+pytorch
1、安装miniconda
资料:http://www.360doc.com/content/23/0124/22/19913717_1064712525.shtml
wget https://repo.miniconda/Miniconda3-latest-Linux-x86_64.sh
sudo chmod +x Miniconda3-latest-Linux-x86_64.sh
sudo ./Miniconda3-latest-Linux-x86_64.sh
2、安装pytorch
资料:https://blog.csdn.net/qq_43728463/article/details/138837943
下载地址:https://pytorch.org/get-started/locally/
- 生成命令后在对应的python环境中执行就可以
- 之前安装的cuda是12.2的,cuda是向下兼容的,这里使用cuda12.1的版本。(不知道如果差太多可不可以,比如用11.8可能也不行)
三、安装issac gym
资料
安装:https://blog.csdn.net/m0_58400126/article/details/144403146
下载地址:https://developer.nvidia.com/isaac-gym/download
效果测试:https://blog.csdn.net/qq_36674060/article/details/145005084
* issac gym停止维护了,这里说可以在18和20用。我实际测试,22也没问题。