貼上我看的教程https://zhuanlan.zhihu.com/p/26871672
一開始想用pycharm裝pytorch,但不知道為什么一直失敗。后來只能conda pip安裝
但conda pip安裝太慢,所以找了鏡像。非常快
conda install pytorch=0.4.0 -c soumith
安裝成功后測試cuda可不可用,torch.cuda.is_available()返回了false
原因是我cuda是8,pytorch1.0.1匹配的是cuda10,所以只能卸載重新下載低版本。
但我卸載又失敗,比較暴躁的直接把環境刪了重新建了一個,再重新裝pytorch0.4.0就可以啦~