深度学习环境安装以及验证(Ubuntu)

深度学习环境安装以及验证(Ubuntu)

前置:CUDA和CUDNN关系,https://blog.csdn.net/xinjieyuan/article/details/127319269
安装参考:https://blog.csdn.net/m0_37605642/article/details/114059682
验证参考:https://blog.csdn.net/jiexijihe945/article/details/131517630

安装补充链接

  • cuda:
    https://developer.nvidia.com/cuda-toolkit-archive
  • cudnn:
    https://developer.nvidia.com/cudnn-downloads
    https://developer.nvidia.com/rdp/cudnn-archive
  • cudnn sample:
    https://github.com/johnpzh/cudnn_samples_v8

验证

cp -r /usr/src/cudnn_samples_v8/ $HOME
cd $HOME/cudnn_samples_v8/mnistCUDNN
make clean && make
./mnistCUDNN

出现Test passed! 验证通过

多版本切换

sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-11.0 1
sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-12.2 2

切换

sudo update-alternatives --config cuda
  • 7
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要在Ubuntu上设置深度学习环境,您可以按照以下步骤进行操作: 1. 安装NVIDIA驱动程序(如果您的系统使用NVIDIA显卡): - 您可以通过运行以下命令来安装推荐的驱动程序版本: ``` sudo apt-get install nvidia-driver-<version> ``` - 安装完成后,通过运行以下命令来验证驱动程序是否正确安装: ``` nvidia-smi ``` 2. 安装CUDA(Compute Unified Device Architecture): - 深度学习框架通常需要使用CUDA进行加速。您可以从NVIDIA官方网站上下载适用于您的显卡和操作系统版本的CUDA安装程序。 - 下载完成后,按照官方文档提供的步骤进行安装。 3. 安装cuDNN(NVIDIA Deep Neural Network library): - cuDNN是一个针对深度学习任务进行优化的GPU加速库。您可以从NVIDIA开发者网站上下载适用于您的CUDA版本和操作系统版本的cuDNN库。 - 下载完成后,按照官方文档提供的步骤进行安装。 4. 安装深度学习框架: - 一旦您完成了NVIDIA驱动程序、CUDA和cuDNN的安装,您可以使用以下命令之一来安装常用的深度学习框架: - TensorFlow: ``` pip install tensorflow-gpu ``` - PyTorch: ``` pip install torch torchvision ``` - Keras: ``` pip install keras ``` - MXNet: ``` pip install mxnet-cu<version> ``` 通过按照以上步骤设置深度学习环境,您就可以在Ubuntu上开始进行深度学习任务了。请注意,具体的安装步骤可能会因为您的系统版本、显卡型号和需要使用的深度学习框架而有所不同,所以请确保参考官方文档并根据您的实际情况进行操作。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值