版權聲明:本文為博主原創文章,歡迎轉載,並請注明出處。聯系方式: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