pytorch安装---快速版

1、用conda 创建一个Python虚拟环境
2、进入这个虚拟环境
3、去pytorch官网:https://pytorch.org/get-started/previous-versions/
找到满足自己Python版本和cuda版本的安装命令
如:

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

注:上面安装时会跳过pytorch 和 torchvision的安装,因为这俩包太大,国内环境受限

4、去这个网址:https://download.pytorch.org/whl/torch_stable.html
下载对应自己cuda版本和Python版本的pytorch和torchvision的whl文件
5、直接用pip install 将第四步下载下来的whl本地安装

你可能感兴趣的:(Pytorch,各种安装)