美文网首页
ubuntu18.04+cuda10.1+cudnn+pytor

ubuntu18.04+cuda10.1+cudnn+pytor

作者: DaiMorph | 来源:发表于2020-03-23 15:08 被阅读0次

    1ubuntu换源

    2开始最容易出错的就是这个地方,参考下图,linux的内核版本一定要对应上,图里面是cuda9.0的,别的版本的在官网可以找到。关于内核版本调整楼主没找到合适的教程,这里直接升级16.04到18.04了。后面很顺利


    image

    后续就可以开始安装cuda和cudnn了,教程参考https://blog.csdn.net/BigData_Mining/article/details/99670642
    非常详细,注意要先卸载本地的英伟达驱动,再重装。这个文章里的双显卡一个是独显一个是nvidia显卡,照着装就可以。
    cudnn:

    sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
     
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
     
    sudo chmod a+r /usr/local/cuda/include/cudnn.h
     
    sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
    

    3anaconda换源
    sudo gedit ~/.bashrc
    修改系统路径

    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --set show_channel_urls yes
    

    关于pytorch安装很慢的问题,一个是系统换清华源,但是楼主用清华源装的pytorch不知道为何一直是cpu版,速度也一般,推荐用豆瓣的源,pip install *** -i https://pypi.doubanio.com/simple/ 速度超快,比如装pytorch和torchvision的时候,用以下两条就可以
    pip install torch -i https://pypi.doubanio.com/simple/
    pip install torchvision -i https://pypi.doubanio.com/simple/

    建议先运行pytorch官网的命令,装以下依赖包,剩下的这两个直接上述命令即可
    conda install pytorch torchvision cudatoolkit=10.1

    相关文章

      网友评论

          本文标题:ubuntu18.04+cuda10.1+cudnn+pytor

          本文链接:https://www.haomeiwen.com/subject/kiswyhtx.html