在安装PyTorch之前要先安装好CUDA、cudNN、以及anaconda,还有就是编译器pycharm,然后你才能开始安装PyTorch。
目录
5、pip list 查看python中安装了哪些第三方库。
1、创建虚拟环境。
- 创建一个PyTorch环境:conda create -n pytorch15 python=3.7
- 激活环境:conda activate pytorch15
- 查看虚拟环境:conda-env list
2、去PyTorch官网。
- 找到之前的版本连接: Previous PyTorch Versions | PyTorch
- 找到PyTorch1.5.0版本,以及对应的cuda10.1
- 然后打开cmd,输入下面的代码:
pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html