安装方法
1、安装Python 3.12.3。
下载地址:https://www.python.org/getit/
2、安装CUDA 12.1。
下载地址:CUDA Toolkit 12.1 Downloads ~| NVIDIA Developer
3、如果以前安装过PyTorch,先卸载以前的PyTorch。
列出安装的PyTorch包:
pip list|findstr torch
卸载所有的PyTorch包:
pip uninstall torchvision torchtext torchdata torchaudio torch
4、由于TorchText对Torch、TorchData、Python版本有严格要求,所以要先安装TorchText。
pip install torchtext --index-url https://download.pytorch.org/whl/cu121
这会安装torch、torchdata、torchtext包。
5、安装其他PyTorch包。
pip install torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
6、查看安装的PyTorch包版本。
pip list|findstr torch
出现:
torch 2.2.0+cu121
torchaudio 2.2.0+cu121
torchdata 0.7.1
torchtext 0.16.2+cpu
torchvision 0.17.0+cu121
常见问题
1、安装包时,可不可以不加:~-~-index-url https:~/~/download.pytorch.org/whl/cu121。
不加~-~-index-url,默认安装的是CPU版本的包,不能使用GPU加速。
2、按照官网方法安装pytorch报错:pip._vendor.urllib3.exceptions.ReadTimeoutError: HTTPSConnectionPool(host='download.pytorch.org', port=443): Read timed out.。
网络问题,参考:常见问题。