ubuntu18.04+cuda10.1+cudnn+pytor
2020-03-23 本文已影响0人
DaiMorph
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
