- 首先CUDA的安装可见tensorflow那篇https://blog.csdn.net/colby_zhang/article/details/107430084
- 然后是pytorch的安装,这里直接进官网就可以了。如果版本太新,想按旧版上面有
install previous versions of PyTorch这句链接,点进去随意挑选。
有个问题就是官网给的安装源都是pytorch官网的,如果需要换源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch