- 使用anaconda安装
- 安装命令查看官网，根据cuda版本等要求选择
- cuda10.1对应的最低版本是1.4.0，安装命令
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch - cuda10.0对应的最高版本是1.2.0，安装命令
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch - cuda9.0对应的最高版本是1.1.0，安装命令
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
欢迎分享，转载请注明来源：内存溢出
微信扫一扫
支付宝扫一扫
评论列表（0条）