1. 輸入命令查詢現(xiàn)有的pytorch版本
import torch
print (torch.__version__) (前后都是兩個(gè)下劃線)
- pytorch 0.1.12-py360
- torchvison :0.1.8-py36_2
2. 更新pytorch和pytorchvision
- cpu版本,適用于python2.7-3.6
- MacOS 不支持 CUDA,如果需要CUDA 8父叙,CUDA 9.0饺汹,或 CUDA 9.2,則需要從源碼編譯安裝
conda install pytorch torchvision -c pytorch
3. 若是安裝過(guò)程中報(bào)錯(cuò):ERROR conda.core.link:_execute(502):
An error occurred while uninstalling package 'defaults::numpy-1.11.3-py36_0'. 說(shuō)明安裝失敗
- 解決方法放棒,輸入以下命令:
sudo conda install pytorch torchvision -c pytorch
然后運(yùn)行后可以看到安裝成功