CUDA9.1版本以及pytorch0.4.0安装
CUDA9.1的安装
百度即可
pytorch0.4.0安装
在pycharm中无法找到对应版本,加入镜像也无法解决问题,下载好pip后可以通过该链接下载CUDA91版本对应的torch。
pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
然后可以更新torchversion版本
pip install torchvision==0.2.2