Windows10安裝cuda、cudnn、pytorch、jupyter、fastai


版權聲明:本文為博主原創文章,歡迎轉載,並請注明出處。聯系方式:460356155@qq.com

一直在Ubuntu中使用pytorch,現在切換到Windows系統。

一、安裝cuda

1、從https://developer.nvidia.com/cuda-downloads下載相應的版本(這里選的是cuda_10.1.105_418.96_win10.exe),安裝,選擇自定義模式。

2、命令行運行:nvcc –V,驗證安裝成功。

二、安裝cudnn

1、從https://developer.nvidia.com/rdp/cudnn-archive下載,但需要注冊,不想注冊的可直接從我的網盤下載(cudnn-10.1-windows10-x64-v7.5.0.56.zip):

  鏈接:https://pan.baidu.com/s/1izpA-aveYbPa-dVUxRhtFw
  提取碼:pfq3

2、解壓后,分別拷貝bin、lib/x64、include中的文件到cuda安裝目錄(如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\)對應目錄下。

三、安裝pytorch

pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp37-cp37m-win_amd64.whl

pip3 install torchvision

四、驗證

import torch

x = torch.rand(5,3)

print(x)

print(torch.cuda.is_available())

五、安裝jupyter

pip install jupyter

命令行運行:jupyter notebook --generate-config,生成配置文件:

  Writing default config to: C:\Users\Administrator\.jupyter\jupyter_notebook_config.py

修改目錄:

  打開這個配置文件,找到“c.NotebookApp.notebook_dir=……”,把路徑改成自己的工作目錄,如:

    c.NotebookApp.notebook_dir = r'D:\fastai\projects'

  設置jupyter:

    pip install jupyterthemes

    jt -l  # 列出背景主題列表

    jt -t grade3 -T # 設置背景主題、顯示toolbar

    jt -h #幫助

    https://www.zhihu.com/question/266988943

六、安裝fastai

1、安裝Microsoft Visual C++ Build Tools,下載http://go.microsoft.com/fwlink/?LinkId=691126,安裝。

2、pip install fastai


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM