安装的时候一定要确保所有的安装包都安装在了一个环境下。mkl,cudatoolkit 一定要确保安装成功,否则 import torch 命令会出现error。由于网速不稳定,我从清华源下载了本地安装包进行安装。由于pytorch1.4以上版本有prune工具包,嗯,真香。
更具体地步骤请参照:另一篇文章《Anaconda 安装+环境配置:pytorch1.5.0+torchvision0.6.0+cudatoolkit10.1》,大同小异啦,只要把对应的版本换一下就可以啦。
安装的时候一定要确保所有的安装包都安装在了一个环境下。mkl,cudatoolkit 一定要确保安装成功,否则 import torch 命令会出现error。由于网速不稳定,我从清华源下载了本地安装包进行安装。由于pytorch1.4以上版本有prune工具包,嗯,真香。
更具体地步骤请参照:另一篇文章《Anaconda 安装+环境配置:pytorch1.5.0+torchvision0.6.0+cudatoolkit10.1》,大同小异啦,只要把对应的版本换一下就可以啦。