自己電腦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版本也能正常跑
推薦閱讀：
