win10 CUDA11.1安装torch1.9 / reformer_pytorch
环境
-
NVIDIA-SMI 457.52
-
Driver Version: 457.52
-
CUDA Version: 11.1
安装torch-gpu
-
conda create -n torch1.9 python=3.8 -
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html -
conda install cudatoolkit=11.1 cudnn=6.0 -
测试torch是否可用
import torch
torch.cuda.is_available()
安装reformer_pytorch
需要先pip install reformer_pytorch
然后执行第一步，安装torch-gpu版本