电脑显卡是p104 8G
Anaconda3-2020.07-Windows-x86_64.exe
cuda_11.1.1_456.81_win10.exe
torch-1.10.1+cu111-cp38-cp38-win_amd64.whl
F:\Anaconda3\envs\keras\Scripts>pip install “torch-1.10.1+cu111-cp38-cp38-win_amd64.whl”
安装完成后测试
F:\Anaconda3\envs\keras\Scripts>python
Python 3.8.3 (default, Jul 2 2020, 17:30:36) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type “help”, “copyright”, “credits” or “license” for more information.
import torch
print(torch.cuda.is_available())
True
print(torch.version.cuda)
11.1
print(torch.version)
1.10.1+cu111