新聞中心
Ubuntu安裝好,顯卡插上去之后,一般的,配置pytorch,tensorflow,mxnet,caffe等深度學(xué)習(xí)環(huán)境需要做三件事情

成都創(chuàng)新互聯(lián)網(wǎng)絡(luò)公司擁有10余年的成都網(wǎng)站開發(fā)建設(shè)經(jīng)驗(yàn),成百上千家客戶的共同信賴。提供網(wǎng)站建設(shè)、網(wǎng)站制作、網(wǎng)站開發(fā)、網(wǎng)站定制、買友情鏈接、建網(wǎng)站、網(wǎng)站搭建、成都響應(yīng)式網(wǎng)站建設(shè)公司、網(wǎng)頁設(shè)計(jì)師打造企業(yè)風(fēng)格,提供周到的售前咨詢和貼心的售后服務(wù)
1.安裝NVIDIA驅(qū)動(dòng)
通過命令行安裝,sudo apt-get install nvidia-375
2. 安裝cuda toolkit
sudo sh cuda_8.0.61_375.26_linux.r7un.26_linux-run
因?yàn)樵诘谝徊揭呀?jīng)安裝NVIDIA驅(qū)動(dòng),所以在安裝cuda的時(shí)候不用安裝opengl,除此之外一路綠燈
3.安裝cudnn
3.1 進(jìn)入下載的cudnn/include, cd cudnn/include, 拷貝cudnn.h 到/user/local/cuda-8.0/include/
sudo cp cudnn.h /user/local/cuda-8.0/include/
3.2 進(jìn)入lib64文件夾,拷貝lib_cudnn_static.a 到/user/local/cuda-8.0/lib64文件夾下,sudo cp lib_cudnn_static.a /user/local/cuda-8.0/lib64/
3.3 拷貝libcudnn.so.5.1.3到/user/local/cuda-8.0/lib64/下,sudo cp libcudnn.so.5.1.3 /user/local/cuda-8.0/lib64/
3.4 進(jìn)入/user/local/cuda-8.0/lib64/下并且創(chuàng)建軟鏈接,sudo ln -s libcudnn.so.5.1.3 libcudnn.so.5, sudo ln -s libcudnn.so.5 libcudnn.so
至此完畢
下面是之前的記錄
分享文章:深度學(xué)習(xí)環(huán)境配置Ubuntu16.04+CUDA8.0+CUDNN5
本文鏈接:http://www.dlmjj.cn/article/cdoejhi.html


咨詢
建站咨詢
