linux通过conda安装cuda和cudnn
时间: 2023-11-28 07:04:46 浏览: 548
Linux下安装cuda
首先,您需要安装Anaconda作为您的Python环境管理器。然后,按照以下步骤安装CUDA和cuDNN:
1. 在NVIDIA的网站上下载适用于Linux的CUDA安装程序,并按照说明进行安装。请注意,您需要选择正确的CUDA版本以与您的显卡兼容。
2. 下载cuDNN库,并按照说明进行安装。请注意,您需要选择与您的CUDA版本兼容的cuDNN版本。
3. 打开终端,创建一个新的conda环境并激活它:
```
conda create --name cuda_env
conda activate cuda_env
```
4. 在新环境中安装PyTorch和其他需要的软件包:
```
conda install pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch
conda install <other_packages>
```
其中,<your_cuda_version>是您安装的CUDA版本。
5. 您现在可以在新环境中运行Python并使用CUDA了:
```
python
import torch
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
```
阅读全文