naconda虚拟环境安装cuda和cudnn
时间: 2023-05-28 15:05:28 浏览: 110
Anaconda虚拟环境中安装cuda
1. 安装cuda
首先需要下载适合自己操作系统的cuda安装包。可以在官网上下载,也可以使用以下命令下载:
```
$ wget https://developer.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
```
下载完成后,使用以下命令进行安装:
```
$ sudo sh cuda_10.1.243_418.87.00_linux.run
```
安装时需要输入一些信息,可以根据自己的情况进行选择。
安装完成后,需要添加cuda路径到环境变量中。可以使用以下命令:
```
$ export PATH=/usr/local/cuda/bin:$PATH
```
或者将以上命令添加到.bashrc文件中,使其永久生效。
2. 安装cudnn
首先需要下载适合自己cuda版本的cudnn安装包。可以在官网上下载,也可以使用以下命令下载:
```
$ wget https://developer.download.nvidia.com/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.1_20200313/Ubuntu18_04-x64/libcudnn7-dev_7.6.5.32-1%2Bcuda10.1_amd64.deb
$ wget https://developer.download.nvidia.com/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.1_20200313/Ubuntu18_04-x64/libcudnn7_7.6.5.32-1%2Bcuda10.1_amd64.deb
```
下载完成后,使用以下命令进行安装:
```
$ sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb
```
安装完成后,需要添加cudnn路径到环境变量中。可以使用以下命令:
```
$ export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
或者将以上命令添加到.bashrc文件中,使其永久生效。
阅读全文