linux下cuda11.4安装pytorch
时间: 2023-05-04 15:01:26 浏览: 184
在Linux上安装CUDA 11.4和PyTorch可以按照以下步骤操作:
1. 首先,安装NVIDIA驱动程序并验证安装是否成功:
sudo apt-get update
sudo ubuntu-drivers devices
sudo apt-get install nvidia-driver-xxx (xxx为安装的驱动版本号)
然后,通过以下命令验证驱动程序是否正确安装:
nvidia-smi
2. 下载和安装CUDA 11.4:
下载CUDA Toolkit 11.4并安装:
sudo sh <download_path>/cuda_11.4.0_xxx.run
在安装期间,请根据提示进行选择。
3. 安装cuDNN 8.2.2:
cuDNN是加速深度学习的重要库之一,需要与安装CUDA的版本匹配。
下载并解压cuDNN 8.2.2文件。然后,运行以下命令安装:
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
4. 安装Anaconda或Miniconda:
下载并安装Anaconda或Miniconda。
5. 创建conda虚拟环境:
conda create --name pytorch_env python=3.8
6. 激活虚拟环境:
conda activate pytorch_env
7. 安装PyTorch:
在conda环境中使用以下命令安装pytorch:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
安装完成后,在conda环境中使用python命令启动Python解释器,然后导入PyTorch准备进行编程和测试。
参考链接:
https://learnopencv.com/installing-pytorch-with-cuda-on-ubuntu-with-anaconda/