利用Anaconda安装PyTorch并将Anaconda创建的虚拟环境添加到Pycharm中

  • 一、利用Anaconda安装PyTorch
  • 1.执行如下的指令查看有哪些环境
  • 2. 创建虚拟环境conda create -n 环境名字(英文) python=x.x(python版本)
  • 3.激活此环境。conda activate 虚拟环境名称
  • 4.安装pytorch-gup版的环境,由于pytorch的官网在国外,下载相关的环境包是比较慢的,所以我们给环境换源。在pytorch环境下执行如下的命名给环境换清华源。
  • 5.进入[PyTorch官网](https://pytorch.org/),然后将下面的命令复制到Anaconda环境下的终端中。
  • 6.安装完成后,配置CUDA的环境变量。
  • 在文件的末尾添加下面两行,注意修改成你的安装路径:
  • 7.测试是否安装成功:
  • (1)终端输入:
  • (2)进入samples的安装目录,选择一个看看是否可以成功编译。
  • 最后出现Test PASSED即可
  • 8.安装cudnn
  • 找到对应自己CUDA的版本进行下载(我的CUDA是10.2版本,cudnn是8.3.2),然后解压。解压完成后执行以下命令:
  • 9.测试
  • 在pycharm创建一个python脚本,在脚本中运行如下代码:
  • 显示如下:
  • 二、将Anaconda创建的虚拟环境添加到Pycharm中


一、利用Anaconda安装PyTorch

1.执行如下的指令查看有哪些环境

conda env list

2. 创建虚拟环境conda create -n 环境名字(英文) python=x.x(python版本)

conda create -n pytorch python=3.8

3.激活此环境。conda activate 虚拟环境名称

conda activate pytorch

4.安装pytorch-gup版的环境,由于pytorch的官网在国外,下载相关的环境包是比较慢的,所以我们给环境换源。在pytorch环境下执行如下的命名给环境换清华源。

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

5.进入PyTorch官网,然后将下面的命令复制到Anaconda环境下的终端中。

anaconda搭建python虚拟环境failed anaconda创建pytorch虚拟环境_pycharm

6.安装完成后,配置CUDA的环境变量。

sudo gedit ~/.bashrc
在文件的末尾添加下面两行,注意修改成你的安装路径:
export PATH="/usr/local/cuda-10.2/bin:$PATH" #这里添加路径出错的话最好加上双引号把路径括起来,这里已经加上了双引号
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH"

7.测试是否安装成功:

(1)终端输入:
nvcc -V

anaconda搭建python虚拟环境failed anaconda创建pytorch虚拟环境_CUDA_02

(2)进入samples的安装目录,选择一个看看是否可以成功编译。
最后出现Test PASSED即可
cd NVIDIA_CUDA-10.2_Samples/0_Simple/vectorAdd
make
./vectorAdd

anaconda搭建python虚拟环境failed anaconda创建pytorch虚拟环境_pytorch_03

8.安装cudnn

cudnn官网

找到对应自己CUDA的版本进行下载(我的CUDA是10.2版本,cudnn是8.3.2),然后解压。解压完成后执行以下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 
sudo cp cuda/lib/libcudnn* /usr/local/cuda/lib64/ 
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

9.测试

在pycharm创建一个python脚本,在脚本中运行如下代码:
import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())
显示如下:

二、将Anaconda创建的虚拟环境添加到Pycharm中

anaconda搭建python虚拟环境failed anaconda创建pytorch虚拟环境_虚拟环境_04


anaconda搭建python虚拟环境failed anaconda创建pytorch虚拟环境_python_05


anaconda搭建python虚拟环境failed anaconda创建pytorch虚拟环境_CUDA_06


anaconda搭建python虚拟环境failed anaconda创建pytorch虚拟环境_python_07


anaconda搭建python虚拟环境failed anaconda创建pytorch虚拟环境_CUDA_08


anaconda搭建python虚拟环境failed anaconda创建pytorch虚拟环境_python_09