1.PyTorch安装(使用Anaconda)
①PyTorch官网选择配置(PyTorch官网)
进入官网,点击Get Started
然后在官网的选择器中选择PyTorch对应的配置信息
Stable版本是PyTorch的稳定版,Preview是未经全面测试的预览版
CUDA版本需要根据显卡驱动版本来选择,官方提供的对照表如下
CUDA Toolkit | Linux x86_64 Driver Version | Windows x86_64 Driver Version |
---|---|---|
CUDA 10.2.89 | >= 440.33 | >= 441.22 |
CUDA 10.1 (10.1.105 general release, and updates) | >= 418.39 | >= 418.96 |
CUDA 10.0.130 | >= 410.48 | >= 411.31 |
CUDA 9.2 (9.2.148 Update 1) | >= 396.37 | >= 398.26 |
CUDA 9.2 (9.2.88) | >= 396.26 | >= 397.44 |
CUDA 9.1 (9.1.85) | >= 390.46 |