通过conda安装pytorch, cuda,cudnn的例子。需要注意的是,通过conda安装的cuda版本按理说要小于等于本机的cuda。
conda install pytorch==0.4.1 cuda92 torchvision==0.2.1 cudatoolkit=9.2 cudnn=7.2.1 -c pytorch
一些关于conda的cuda和本机cuda的讨论:
https://www.cnblogs.com/yhjoker/p/10972795.html
版权声明:本文为wz22881916原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。