windows下安裝pytorch1.7.1
背景
由於課題需要pytorch,所以轉向pytorch。而網上的都下載特別慢,尤其是官網上面的,如果是用conda下載的話,清華源的鏡像會特別不好使,但是如果用官網的話,下載會直接報錯:
而官網上的要求安裝的包如下:
但是咱們最后需要的包也就是torch、torchvision。所以直接安裝這兩個就可以了。
操作:
點擊上面連接,找到這個:
注意:cu110指的是cuda版本是11.0版本的,cp37指的是python=3.7版本,win_amd64指的是windows版本64位。下載好了之后,在目錄處輸入cmd,pip install 下載的文件名。
安裝成功后:
強烈建議提前使用pip安裝numpy,安裝太慢了。
同理再安裝torchvision:
最后測試: