conda+CUDA+pytorch环境搭建

一、Anaconda下载：https://www.anaconda.com/download#

      检查是否安装好

      

二、CUDA安装

1.CUDA的下载地址： CUDA Toolkit 12.4 Downloads | NVIDIA Developerhttps://developer.nvidia.com/cuda-downloads

        2.下载对应的版本。

3.安装cudnn

cudnn就是一个神经网络加速包，也谈不上安装，把包复制进去就好了。

按需求下载cudnn的安装文件：https://developer.nvidia.com/rdp/cudnn-archive

解压下载的文件，可以看到cuda文件夹，在当前目录打开终端，执行如下命令：

4.检测是否安装成功

在CMD输入nvcc -V，出现版本号表示安装成功。

三、pytorch的安装

1.pytorch和CUDA有版本问题，查看CUDA版本。cudatoolkit的版本不能高于cuda driver api的版本，用nvidia-smi可查看最高版本。

2.下载对应的版本  官网：PyTorch

找到对应的版本后，复制红框内的内容使用conda进行下载。这里注意，如果下载时间过长或者下载失败，可以使用清华镜像源进行下载，地址：Index of /anaconda/cloud/pytorch/linux-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

这样下载成功后，直接把下载好的包放进环境中即可。

3.检查是否安装成功

进入python命令行后，执行以下命令。如果不报错就说明安装成功了。