526互联

win10 CUDA11.1安装torch1.9 / reformer_pytorch

发布时间 2023-08-25 20:21:10作者: 小陈同学爱编程

环境

  • NVIDIA-SMI 457.52

  • Driver Version: 457.52

  • CUDA Version: 11.1

安装torch-gpu

  1. conda create -n torch1.9 python=3.8

  2. 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

  3. conda install cudatoolkit=11.1 cudnn=6.0

  4. 测试torch是否可用

import torch
torch.cuda.is_available()

安装reformer_pytorch

需要先pip install reformer_pytorch
然后执行第一步，安装torch-gpu版本