import torch.nn.functional as torch_F
windows linux通用:
1.7.1 支持cuda10.1
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
错误安装方法:
pip install torch==1.7.1+cu101
可以去官网https://pytorch.org/
直接选择需要的版本,进行下载:
https://pytorch.org/get-started/previous-versions/
下载历史版本whl自己安装,可以选cuda版本:
https://download.pytorch.org/whl/torch_stable.html
conda
install
mkl cffi
下载mkl
https://pypi.python.org/pypi/mkl
下载:cffi
https://pypi.python.org/pypi/cffi
conda install --offline pytorch-0.1.12-py36_0.1.12cu80.tar.bz2 #离线安装,这个版比较低
这个版本是:0.1.12,我们需要安装0.3.1
有一种简单方法:直接把pytorch-0.3.1-py36_cuda80_cudnn6he774522_2\Lib\site-packages拷贝到
D:\Users\Administrator\Miniconda3\envs\python3\Lib\site-packages
即可安装成功,参考caffe安装方式