ubuntu22.04安装cuda和pytorch

本台服务器已经安装：

pycharm2024

andaconda3

python3.12

安装cuda前，需要安装nvidia驱动，自行查阅安装流程 

1.先检查nvidia驱动，和能安装的cuda最高版本 

nvidia-smi

 

 cuda最高为12.2

如果你没有更新过显卡驱动，需要更新成为最新版本的显卡驱动。如果已经更新过了就无视。显卡驱动版本决定了你能装的的CUDA最高版本是多少。而CUDA版本又决定了你装的pytorch最高版本是多少。

 安装pytorch，首先要决定你要安装的pytorch版本需要的cuda版本是多少，当然也可以先安装cuda版本在去寻找合适的pytorch版本。 本文采用第二种。

 2.安装CUDA的版本

进入Pytorch：Pytorch  

 

打开pytorch的网站，然后我们打算安装12.1。我们下面将演示安装CUDA12.1的流程。 注意这里选择安装cuda11.8也是完全没有问题呢，假如你的最新驱动最高能安装的cuda版本只有cuda12.0，那么你只能选择安装cuda11.8。

进入cuda网站 

进入CUDA：CUDA  

点击然后进入，选择要下载的类型 

 

进入控制台终端， 输入下面代码

 

 

输入accept

 

将X取消，按空格取消，已经安装过驱动了。这里不需要。然后我们选择install 回车  

 

如果有这种情况， 直接在安装命令后加  “--override”

 

ok安装完成。但要配置一下文件

vim ~/.bashrc

 按”i“进入编辑模式，输入下面代码，再按“：wq”退出

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/extras/CPUTI/lib64
export CUDA_HOME=/usr/local/cuda/bin
export PATH=$PATH:$LD_LIBRARY_PATH:$CUDA_HOME

更新配置文件 

source ~/.bashrc

 最后检查下配置是否正确

nvcc -V

 

配置成功

3.在Pycharm中安装Pytorch 

进入控制台，输入 之前选cuda时候的pytorch代码

 

 

安装过程很慢，最后根据提示输入y就完成了