自己电脑cuda driver版本是11.1的情况下 如何装pytorch的gpu版本？
自己电脑cuda driver版本是11.1的情况下 如何装pytorch的gpu版本（因为pytorch只支持到10.2的cuda版本）
按照官网的命令安装就可以了，之前没有11.0的时候可以按照源码编译
pip install torch===1.7.0+cu110 torchvision===0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
如果觉得环境搭建麻烦，推荐一个小众的gpu云featurize.cn
建议从源码编译
pytorch 1.7版本支持CUDA11
我也是cuda11 创建虚拟环境 conda下载 不需要手动装cuda和cudnn 自动会下载pytorch版本对应的cuda版本包 一个月前亲测 （不过现在1.7支持cuda11了 还没试过）
去pytorch官网在get started里面选好对应版本和安装方式，直接复制生成的命令安装就行了
我的是
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:54:10_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.relgpu_drvr455TC455_06.29190527_0
直接用
PyTorch?pytorch.org
这个链接里面你就选11.0，然后pip install 在pycharm中创建的virtual环境里可以运行的
我用的10.2版本也能正常跑
推荐阅读：
