Linux小白讲述ubuntu18.04 安装cuda和pytorch的注意事项
Linux小白讲述ubuntu18.04 安装cuda和pytorch的注意事项1. cuda安装2.cudnn安装3.pytorch安装4.总结
1. cuda安装
1.1 版本问题
务必要先从pytorch官网看对应的cuda和cudnn的版本,我第一次安装时听别人讲CUDA比较难装,所以就打算先把硬骨头啃了,直接去cuda官网下载了最新的10.2版本,结果在安装pytorch时发现这个包里c...
原创
2020-03-14 12:52:05 ·
800 阅读 ·
0 评论