pytorchのインストールとアンインストール

pytorchのインストールとアンインストール

1つ、pytorchのインストール

gpuバージョンのtensorflowをインストールしました

1.1最初にcmdを開いて、自分でインストールしたcudaのバージョンを確認します

入る

nvcc --version

ここに画像の説明を挿入
cuda10.0であることがわかります

1.2オフラインダウンロードパッケージを見つけてインストールします

pytorchオフラインダウンロードアドレス:pytorchオフラインダウンロード

Windowsシステム+ Pythonバージョン-cudaバージョンに従ってダウンロード
ここに画像の説明を挿入

1.3pipを使用してインストールする

pip install torch-1.0.0-cp36-cp36m-win_amd64.whl

トーチ1.0.0に対応するトーチビジョンのバージョンは0.2.0です。

pip install torchvision==0.2.0

1.4GPUバージョンが正常にインストールされていることを確認します

テスト手順は次のとおりです。

import torch

print(torch.__version__)
print(torch.cuda.is_available())

演算結果:

1.0.0
True

PyTorchのバージョンは1.0.0Trueです
。これは、GPUバージョンが正常にインストールされたことを意味します。

2、pytorchのアンインストール

1.condaを使用してPytorchをアンインストールします

conda uninstall pytorch
conda uninstall libtorch

2.pipを使用してPytorchをアンインストールします

pip uninstall torch

pytorchがcondainstallコマンドでインストールされた場合は、最初の方法を使用します。pipinstallコマンドでインストールされた場合は、2番目の方法を使用します。または、両方の方法を試してください。

おすすめ

転載: blog.csdn.net/mao_hui_fei/article/details/112078113