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
网友评论