快速安装Pytorch以及CUDA匹配问题

上传:gqw98975 浏览: 11 推荐: 0 文件:PDF 大小:76.87KB 上传时间:2020-12-31 07:21:37 版权申诉
pytorch的安装按照官网上来操作是最快速也最稳妥的方式。https://pytorch.org/get-started/locally/ 官网非常贴心的给出了各种安装方式,满足各种需求 首先关于python的版本可参见pytorch的github: 需要先安装版本匹配的CUDA以及cuDNN,具体的安装命令如下图操作: 根据自己的需求选择合适的版本,网站就会给出相应的安装命令,目前来看支持CUDA最好的版本就是9.2和10.1 例如上图的选择会返回安装命令: conda install pytorch torchvision cudatoolkit=10.1 -c pytorch 在W
上传资源
用户评论