Install Anaconda + NVIDIA graphics card driver + pytorch under win10_ gpu

1. anaconda install

Go to Official website Can be installed .

After installation , We can modify it conda Source ：

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

After modification , Check whether the modification was successful ：

conda config --show channels

2. Driver.

1、 Uninstall the original graphics card driver

2、 Official website Installing a new graphics card driver

After installation , Right click on the desktop to see the graphics card driver ,

3. pytorch_gpu install

1、 open anaconda prompt

2、 Creating a virtual environment

conda create -n pytorch python=3.6

Enter the following command to check whether the creation is successful ,

conda info --envs

As shown in the figure , That is, the creation is successful .

3、 Activate the environment

activate pytorch

4、 Official website install pytorch

Because the server is abroad , So the download will be slow . We can download through Tsinghua source , This will greatly speed up the download , The command is modified as follows ,

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio===0.10.2+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

After downloading , We can verify it ,

activate pytorch
python
import torch
print(torch.cuda.is_available())
print(torch.__version__)