剛開始深度學習的入門，都推薦在linux環境下搭框架，先入門學習，就還是在windows下試試.
顯卡是英偉達 NVIDIA GTX 960M,配置不高

anaconda選擇的是 anaconda3
python是3.7.6

下面就是pytorch的安裝。

前面的一些過程可以在B站或者其他的教程里找到。
在我進行搭環境時，最煩的是使用conda安裝pytorch的緩慢，甚至直接中斷，折騰了三四個小時.
后來使用清華鏡像源進行安裝，流暢舒服，十分鐘搞定。
在anaconda的prompt中輸入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

將源換為清華鏡像源。
最后運行下面的命令就可以快速安裝好

conda install pytorch torchvision cudatoolkit=10.1

不過我在安裝完，進行驗證時報錯： The NVIDIA driver on your system is too old (found version 9020). Please update your GPU driver by downloading and installing a new version from the URL:
這里需要更新顯卡驅動
去鏈接: 顯卡驅動.根據自己的顯卡系列尋找最新的顯卡驅動，只需要找到自己顯卡的型號，其他不需要改動，按照提示下載安裝顯卡驅動。
驗證語句

import torch print(torch.cuda.is_available())

驗證語句也是在prompt中如下圖：返回true,代表GPU版可用，否則出現錯誤

總結

以上是生活随笔為你收集整理的Win 10 GTX 960下 Pytorch的安装与验证的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。