0
点赞
收藏
分享

微信扫一扫

cuda安装教程

佃成成成成 2022-02-16 阅读 74

1、安装cuda
sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.105-418.39_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-1-local-10.1.105-418.39/7fa2af80.pub
sudo apt-get update
sudo apt-get install -y cuda
sudo gedit ~/.bashrc(写入export PATH=$PATH:/usr/local/cuda-10.1/bin
export LD_LIBRARY_PATH=:/usr/local/cuda-10.1/lib64)
source ~/.bashrc
在这里插入图片描述2\安装cudnn
安装运行时库:
sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb

安装开发者库:
sudo dpkg -i libcudnn8-dev_7.6.5.32-1+cuda10.1.deb

安装代码示例和cuDNN库文档:

sudo dpkg -i libcudnn8-doc_7.6.5.32-1+cuda10.1.deb

测试
cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN

如果cuDNN正确安装,将会打印出如下信息:
Test passed!

3\安装tensorrt

举报

相关推荐

0 条评论