赞
踩
27. GPU版本-安装 PyTorch(方法2)_哔哩哔哩_bilibili
命令行查看本设备的cudaDriver版本
nvidia-smi
查看本驱动能够支持最大版本cuda:
打开cmd
conda activate pytorchName
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
复制下载地址,通过迅雷快速下载到本地
cmd下载
(pytorch) C:\Users\MyLENOVO>pip install address
下载结束后再次执行
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
下载剩余的包
下载成功即可完成
进入pytorch环境,
- (pytorch) C:\Users\LENOVO>python
- >>> import torch
- >>> torch.cuda.is_available()
- True
完成
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。